python - 我是否使用“rescale=1./255”？构建VGG-like CNN

Question

我正在构建一个与 VGG16 具有相似架构的 CNN。我知道在 VGG16 中你确实重新调整了你也以 0 为中心。现在我从头开始训练我的，这有什么不同吗？我应该这样做还是不这样做？有谁能帮忙吗？

datagen = ImageDataGenerator(rescale=1. / 255)

非常感谢你们！<3

Answer 1

在训练神经网络时，通常重新缩放到 0-1 总是一个不错的选择，所以是的。这背后的原因是，当输入归一化时，神经网络往往会产生更好的结果。您可以使用像素重新缩放和不进行像素重新缩放运行相同的精确实验，并亲自查看结果！

Answer 2

最好重新调整像素值，使平均值为 0，值的范围为 1 到 -1。大多数预训练模型都要求您这样做。您可以使用 tf.keras.applications.imagenet_utils.preprocess_input 来实现。文档在这里。