人工智能学习07--pytorch02--数据集取图片+以图片名为文件名的txt文档(label)
1、取图片fromtorch.utils.dataimportDataset#importcv2fromPILimportImageimportosclassMyData(Dataset):#help()中说,想要使用Dataset,需要重写。__getitem__和__len__def__init__(self,root_dir,label_dir):#初始化函数#可以把self指定的变量给后面