【发布时间】:2020-08-04 14:17:38
【问题描述】:
我想对 ResNet50 ImageNet 预训练模型进行微调,关于微调的图像预处理有几个问题。
在 ImageNet 预处理中,我们需要减去像素的平均值([103.939, 116.779, 123.68])。当我使用我的数据集进行微调时,我应该减去 ImageNet 的平均值还是减去我的数据的平均值。
我确实看到很多人将数据重新缩放到 [0,1],但是预训练模型 (ImageNet) 使用 [0,255] 中的图像缩放。人们为什么这样做?合理吗?
【问题讨论】:
-
我想如果你使用 imagenet 的平均值,你会得到最好的准确度。但是,我认为这不会有很大的不同。如果你做了一个实验并让我们知道,那就太酷了!当您有许多不同比例的不同特征时,通常使用 0 到 1 之间的比例。对于图像分类,这不是必需的,只需确保所有图像都具有相同的缩放比例,并且模型是在具有相同缩放比例的图像上训练的。
标签: python deep-learning