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))
计算数据集的均值和标准差及归一化后的数据变成图片
最新推荐文章于 2025-02-03 11:01:28 发布