Skip to content

dataset

desperadoccy edited this page May 28, 2024 · 2 revisions

dataset模块介绍

dataset模块用来管理数据集,可以自定义数据集类,以创建多个数据集,构成多模态场景。 提供获取训练集、测试集和生成数据分布等功能。

  • path: 该数据集类的路径
  • params: 初始化该类所需的params

现有类介绍

本节下介绍的参数填写在params字段中

  • CIFAR10
  • CIFAR100
  • MNIST
  • FashionMNIST
  • EMNIST
  • PAMAP2
  • SVHN
  • UCIHAR

BaseDataset

该类为基类,定义了必须实现的函数,以及基本属性

Clone this wiki locally