FCNを大きな画像サイズでテストする

Question

CaffeでFCN-VGG16ネットワークをテストしています。入力画像のサイズを固定サイズに変更する代わりに、毎回ネットワークを画像サイズに変更します。私はmatcaffeとpycaffeの両方でこれを試しました。どちらの場合も、小さな画像（たとえば、500x500）で実行できるようです。しかし、大きなサイズの画像（1920 x 1080など）を使用すると、「Check failed：error == cudaSuccess（2対0）」というエラーが表示されます。私は12 GBのメモリを搭載したQuadro M6000 GPUでこれを実行しています。どんな助け/助言も高く評価されます。

Answer 1

私はそれがメッセージから分かりますが、の場合はに入力サイズを減らす必要があります.GPUには、それを供給するための十分なオンボードメモリがありません。バッチサイズを減らすことはできますか？そうでない場合、本当には2Mピクセル全体を必要としますか？ほとんどのモデルは、切り抜かれたイメージや縮小されたイメージではうまくいきます。これを（CPU上の）入力処理の一部として行い、任意のサイン入力に対してパラメータ化します。

ネットワークのサイズを変更すると、特にすべてのレイヤーをサイズ変更し、入力サイズでスケーリングすると、メモリが高価になります。あなたが与えた場合、これはあなたのメモリ要件を約8倍向上させます。

サイズの確認として、ネットワークの初期化時にメモリの数字を見てください。私はCaffeがメモリ要件をレイヤーで報告していると信じています。