qq_43398933的博客

转载 Rsesnet 残差网络

import torchimport os,globimport randomimport csvfrom torch.utils.data import Datasetfrom PIL import Imagefrom torchvision import transformsfrom torch.utils.data import DataLoader# 第一部分：通过三个步骤得到输出的tensor类型的数据class Dataset_self(Dataset): .