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