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))
计算数据集的均值和标准差及归一化后的数据变成图片
最新推荐文章于 2024-06-26 10:13:29 发布
本文介绍了如何在Python中利用PyTorch库处理深度学习数据集,详细阐述了计算数据集的均值和标准差的过程,并展示了如何将数据归一化,以便更好地应用于神经网络模型。同时,还讨论了归一化后的数据如何转换为可视化的图片,以辅助理解和调试模型。
摘要由CSDN通过智能技术生成