画像の不要物を削除 AOT-GAN-for-Inpainting

Inpaintingやってみたいけど難しそう

意外と簡単です。

使い方

InpaintingモデルAOT-GAN-for-Inpaintingの使い方です。

リポジトリからトレーニング済みの重みをダウンロードしておきます。
リポジトリをクローンして、

・トレーニング済みの重みのパス
・画像ディレクトリ
・マスク画像ディレクトリ

を指定して実行します。
マスク画像は入力画像と同じ大きさの白黒（GrayScale）画像で、隠したいところを白く塗ります。

# clone
git clone https://github.com/researchmm/AOT-GAN-for-Inpainting.git
cd AOT-GAN-for-Inpainting/src

# run
python test.py --pre_train G0000000.pt --dir_image ../examples/object/image --dir_mask ../examples/object/mask

outputsディレクトリに出力が保存されます。

インプット画像

マスク画像

出力