计算数据集的均值和标准差及归一化后的数据变成图片

imagelist = os.listdir(r"D:\cat_dog\img")
data = MyDataset(r"D:\cat_dog\img")
#数据集的均值和标准差
# loader = DataLoader(dataset=data,batch_size=12000,shuffle=True)
# data= next(iter(loader))[0]
# mean = torch.mean(data, dim=(0,2,3))
# std = torch.std(data, dim=(0,2,3))
# print(mean ,std)
#归一化后的数据变成图片
x 

你可能感兴趣的:(深度学习,python,深度学习,python,人工智能,dataset,pytorch)