0

私はU-netアーキテクチャでセグメンテーションモデルを訓練しています。入力画像のサイズは250x250です。Caffeセグメンテーションネットワークの出力サイズを入力と同じにする方法は?

現在のところ、畳み込みレイヤーのいくつかのパディングを手動で調整して、モデル出力が同じサイズ(250x250)であることを確認しています。

しかし、別のサイズの画像、たとえば500x500の画像を入力すると、出力サイズは506x506になります。

出力サイズがすべてのサイズの入力と同じになるようにするにはどうすればよいですか?

答えて

1

"Crop"レイヤーを使用すると、出力の形状を強制的に同じにすることができます。
Uネットでは、パディングエラーの蓄積を避けるために、アップサンプリングのたびにクロップレイヤーを使用することをお勧めします。

「パディングエラー」について:
入力が100x100の場合、2倍の係数で3回サンプリングすると13x13のフィーチャマップになります。今
、あなたは三回アップサンプリング×2ずつ

13x13 --> 26x26 --> 52x52 --> 104x104 

はあなたが原因パディング/丸め(あなたの質問にあなたは6を持っている)に追加された「追加」4つのピクセルを持っている場合。しかし

13x13 --> 26x26 \crop 25x25 --> 50x50 --> 100x100 

あなたが最初のアップサンプリング後に非自明な作物があることがわかり、それが唯一の4

をそのレベルで1つのピクセル誤差を持っている、としませアップサンプルそれぞれの後に、あなたの場合 "Crop"
+0

パディングエラーは正確にはどういう意味ですか? サイズに合わせて最後のアップサンプリングレイヤの後にクロップレイヤーを追加したところ、正常に動作しているようです。それに問題はありますか? – shubhamgoel27

+0

@ shubhamgoel27私の編集を参照してください – Shai

+0

編集をありがとう、どのようなサイズで中間層でトリミングする必要がありますか? (私の入力サイズが固定されていないと仮定して) – shubhamgoel27

関連する問題