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

深度学习项目四:创建自定义数据集的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基本类需

你可能感兴趣的:(深度学习和视觉项目实战,深度学习,人工智能,python,pytorch)