深度学习项目四:创建自定义数据集的dataset, DataLoader

深度学习项目四:创建自定义数据集的dataset, DataLoader



前言

在构建神经网络结构模型后, 需要对数据进行输入训练迭代,就需要构建数据集的dataset, DataLoader。如果使用数据集是torch内置数据集,可直接继承torch.utils.data的dataset, DataLoader从而完成训练,评估。
本次项目主要讲解对于自己的数据集,构建自定义数据集的dataset, DataLoader。


一、构建dataset基本类

  • 构建datase基本类需

猜你喜欢

转载自blog.csdn.net/m0_60890175/article/details/131749936