PyTorch 中推荐的标准化参数背后的逻辑是什么?

数据挖掘 深度学习 火炬 火炬视觉
2022-03-09 02:06:38

在 torchvision.models 的 PyTorch文档中,规定图像必须在 [0,1] 范围内加载,然后使用mean = [0.485, 0.456, 0.406]和进行标准化std = [0.229, 0.224, 0.225]这些特定值背后的逻辑是什么?

1个回答

您可以在此处此处的 stackoverflow以及pytorch 论坛中找到答案

如链接中所述,这些值是根据 Imagenet 训练数据计算的。从最后一个链接,这是Soumith Chintala的解释:

在 Imagenet 上,我们对数据集进行了传递并计算了每个通道的均值/标准差。