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)

`