我有一个 CNN,它在我创建的数据集上学习得很好。我向该网络添加了批量标准化以尝试提高性能。
但是..当我尝试对单个图像进行预测时,我总是得到相同的结果(无论图像如何)。我认为这是因为我需要批次来实际进行批次标准化。
那么是否可以使用 BN 使用 CNN 对单个图像进行预测?我想在我的网络完成训练后删除 BN 层,这是要走的路吗?
谢谢 :)
我有一个 CNN,它在我创建的数据集上学习得很好。我向该网络添加了批量标准化以尝试提高性能。
但是..当我尝试对单个图像进行预测时,我总是得到相同的结果(无论图像如何)。我认为这是因为我需要批次来实际进行批次标准化。
那么是否可以使用 BN 使用 CNN 对单个图像进行预测?我想在我的网络完成训练后删除 BN 层,这是要走的路吗?
谢谢 :)
我在这里找到了确切的答案和我面临的问题:https ://r2rt.com/implementing-batch-normalization-in-tensorflow.html 在“使用模型进行预测”中解释说,在使用 BN 时,在训练期间您需要估计训练集上的总体均值和总体方差的时间,这样您就不必在进行测试时使用批处理(这将是“作弊”):)