バッチサイズ変更時のTensorflowオブジェクト検出パッケージのエラー

Question

テンソルフロー用に新しくリリースされたオブジェクト検出APIを試しています。 チュートリアルでサンプルトレーニングプログラムを使用しました。つまり、ペットデータセットでFRCNN-Resnetモデルを微調整しました。 1つのGPUだけを使用すると、バッチサイズを1以上の値に変更すると、エラーメッセージが常に表示されます（デフォルトは1）。

InvalidArgumentError（トレースバックについては上記を参照）：ConcatOp：入力の寸法は、shape [0] = [1,750,600,3]対[1] = [1,600,804,3]

と一致する必要があります。

Answer 1

keep_aspect_ratio_resizerを使用すると、各画像のサイズが異なるため、バッチサイズ1でしかトレーニングできない可能性があります。バッチサイズを大きくするために、現在APIでこれを処理する唯一の方法は、 fixed_shape_resizerを使用してください。この例については、SSDの設定の一部を参照してください。