it__

mode,fileName):#初始化函数super(Pokemon,

rootself.resize

sorted(os.listdir(path)):subDir

os.path.join(root,

os.path.isdir(subDir):continueelse:self.name2label[name]

len(self.name2label.keys())csv_path

fileName)print(\n

os.path.exists(csv_path):images

self.load_images()self.save_csv(fileName,

images)self.images,

self.load_csv(fileName)self.divide_data(mode)

load_images

self.name2label.keys():#pokeon\\newtwoo\\00001.png#返回所有匹配的文件路径列表。

它只有一个参数pathname定义了文件路径匹配规则这里可以是绝对路径也可以是相对路径。

下面是使用glob.glob的例子pngPath

name,*.png)jpgPath

,len(images))random.shuffle(images)return

images

self.name2label[name]writer.writerow([img,

load_csv

int(label)images.append(img)labels.append(label)m

len(images)n

self.images[start:end]self.labels

len(self.images

torch.tensor(mean).unsqueeze(1).unsqueeze(1)std

torch.tensor(std).unsqueeze(1).unsqueeze(1)x

x_hat*stdmeanreturn

x:Image.open(x).convert(RGB),transforms.Resize((int(self.resize*1.25)

transforms.ToTensor(),transforms.CenterCrop(self.resize),transforms.Normalize(mean[0.485,

0.456,

batchY.shape)viz.images(db.denormalize(batchX),nrow8,

winbatchX,optsdict(titlecurtime))viz.text(str(batchY.numpy()),winbatchY,optsdict(titlelabel))time.sleep(10)if

__name__

normalizetransforms.Normalize(meanimgMean,stdimgStd)

transformtransforms.Compose([transforms.RandomCrop(180),transforms.RandomHorizontalFlip(),transforms.ToTensor(),

#将图片转换为Tensor,归一化至[0,1]normalize

])datasetImageFolder(./data/train,transformtransform)参考

torchvision.datasets.ImageFolder使用详解_☞源仔的博客-CSDN博客

课时102