用imagefold对数据集进行加载,如何计算图片归一化的均值和方差
时间: 2024-03-26 16:36:52 浏览: 187
计算图像均值和方差
5星 · 资源好评率100%
使用imagefold加载数据集时,可以使用torchvision.transforms.Compose进行图片的预处理,其中torchvision.transforms.Normalize用来进行归一化处理,需要传入均值和方差。计算图片的均值和方差可以使用torchvision.transforms.Normalize的另一个函数torchvision.transforms.Normalize.calculate_mean_and_std来计算。
具体实现步
阅读全文