Inpaintingモデル DeepFillv2の使い方

方法

インストール

git clone https://github.com/csqiangwen/DeepFillv2_Pytorch.git
cd DeepFillv2_Pytorch
pip install imgcrop
gdown https://drive.google.com/u/0/uc?id=1uMghKl883-9hDLhSiI8lRbHCzCmmRwV-

推論

python3 test.py --baseroot './test_data/' --baseroot_mask './test_data_mask/' --results_path './results' --gan_type 'WGAN' --gpu_ids '1' --epoch 40 --batch_size 1 --num_workers 8 --pad_type 'zero' --activation 'elu' --norm 'none'

🐣

---

フリーランスエンジニアです。
お仕事のご相談こちらまで
[e-mail]

機械学習、ARアプリ（Web/iOS）を作っています。
機械学習／AR関連の情報を発信しています。

Twitter
Medium
GitHub