pytorch code导入数据集(训练集、测试集)

• 代码和数据文件夹处于平行路径下
  

• 得到path里面一层的所有文件和文件夹
  os.listdir(path)
partimgs = os.listdir(trainingset_path+imgFoldName) 即'NCT-CRC-HE-100K/'+'ADI'

• 导入包
  from torch.utils.data import Dataset, DataLoader
  pytorch文档链接
  

-例子
`
class testSet(Dataset): #定义测试集
def init(self):
super(testSet, self).init()

def getitem(self, index):
image = cv2.imread(testset_path+fold[int(labels1[index])]+'/'+imgs1[index])
#训练集的话这边有一些数据增强的操作，旋转啥的
image = np.transpose(image, (2,0,1))
image = image / 255
image = torch.tensor(image, dtype=torch.float32)
return image, labels1[index]
def len(self):
return len(imgs1)
`