6 Pytorch加载数据初认识

一、数据加载的两个类

  • Dataset
  • Dataloader
    image

二、查看数据集和label

  • 文件的名称就是label

image

  • 训练的图片和训练的label是分开的

image

image

  • 直接把label写在图片名上

三、数据模块

  • 引入数据模块
from torch.utils.data import Dataset
  • 查看Dataset的功能
#两种
help(Dataset)
Dataset??

image

image

posted @ 2022-05-09 11:57  Trouvaille_fighting  阅读(45)  评论(0)    收藏  举报