在 torchvision.models 的 PyTorch文档中,规定图像必须在 [0,1] 范围内加载,然后使用mean = [0.485, 0.456, 0.406]和进行标准化std = [0.229, 0.224, 0.225]。这些特定值背后的逻辑是什么?
PyTorch 中推荐的标准化参数背后的逻辑是什么?
数据挖掘
深度学习
火炬
火炬视觉
2022-03-09 02:06:38
1个回答
您可以在此处和此处的 stackoverflow以及pytorch 论坛中找到答案
如链接中所述,这些值是根据 Imagenet 训练数据计算的。从最后一个链接,这是Soumith Chintala的解释:
在 Imagenet 上,我们对数据集进行了传递并计算了每个通道的均值/标准差。