2017-06-28 16 views
1

私は、作成したデータセットで非常にうまく学習しているCNNを持っています。このネットワークにバッチ正規化を追加してパフォーマンスを向上させました。バッチ正規化を使用した場合の単一予測

しかし、私が単一の画像で予測をしようとすると、いつも同じ結果(画像が何であれ)で終わります。実際にバッチの正規化を行うにはバッチが必要なためだと思います。

BNを使用してCNNで1つの画像を予測することは可能ですか? 私のネットワークがトレーニングを終えたら、BNレイヤーを削除することを考えました。

ありがとうございました:)

+0

これはバッチの正規化による問題ではありません。より詳細な情報を提供してください。 –

+0

実際にバッチの正規化が原因だと思うのは、バッチの正規化を行うと、一枚の画像(一枚ではなく)のネットワークで予測が行われるからです。しかし、私は1枚の画像を予測する予測をしたいと思います。 –

+0

私はモデルでこのような動作を再現することはできません。彼らは常にBNと単一の画像バッチでうまくいきました。また、テスト時には、正規化統計は固定され、バッチでは計算されません。これはすべて、テクニックではなくコード内の問題を指しています。 –

答えて

1

を私は正確な答えと私はここに直面している問題が見つかりました:「モデルとの予測を行うことは、」BNを使用した場合、訓練期間中にあなたが推定する必要があると説明されている中https://r2rt.com/implementing-batch-normalization-in-tensorflow.html をあなたのトレーニングセットの人口平均と母集団の分散は、テストを行うときにバッチを使用する必要はありません(「不正行為」になります)。

関連する問題