深度学习项目四：创建自定义数据集的dataset, DataLoader
文章目录
- 深度学习项目四：创建自定义数据集的dataset, DataLoader
- 前言
- 一、构建dataset基本类
- 二、创建train_dataset,test_dataset及相应DataLoader
- 1.数据集的index创建
- 2.创建自定义dataset类
- 3.划分训练集和测试集，创建相应的DataLoader
- 三、全部代码
前言
在构建神经网络结构模型后, 需要对数据进行输入训练迭代,就需要构建数据集的dataset, DataLoader。如果使用数据集是torch内置数据集,可直接继承torch.utils.data的dataset, DataLoader从而完成训练，评估。
本次项目主要讲解对于自己的数据集，构建自定义数据集的dataset, DataLoader。
一、构建dataset基本类
- 构建datase基本类需
)
