-
代码和数据文件夹处于平行路径下
-
得到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)
`