ランキングに入る高品質モデル
Paper with code places2インペインティング部門でLamaに次ぐ第4位に入るモデル(2023年1月現在)。
使い方
インストール
git clone https://github.com/MADF-inpainting/Pytorch-MADF.git
cd Pytorch-MADF
リポジトリから事前トレーニング済みモデル・ウェイトをダウンロードします。
laces2 Celeba PSVの3モデル(いずれも512px*512px)が提供されています。
データ
以下のような画像とマスクのリストファイルを用意します。
place2_example_list
examples/places2/case1.png examples/places2/case1_mask.png
examples/places2/case2.png examples/places2/case2_mask.png
examples/places2/case3.png examples/places2/case3_mask.png
examples/places2/case4.png examples/places2/case4_mask.png
examples/places2/case5.png examples/places2/case5_mask.png
examples/places2/case6.png examples/places2/case6_mask.png
examples/places2/case7.png examples/places2/case7_mask.png
python test.py --list_file places2_example_list --snapshot places2.pth
resultに結果が保存されます。
🐣
フリーランスエンジニアです。
お仕事のご相談こちらまで
rockyshikoku@gmail.com
機械学習、ARアプリ(Web/iOS)を作っています。
機械学習/AR関連の情報を発信しています。