比較的大きな穴でも修復できる inpaintingモデルRecurrent Feature Reasoning for Image Inpainting

正解→マスクした画像→推論結果

使い方

インストール

git clone https://github.com/jingyuanli001/RFR-Inpainting.git
cd RFR-Inpainting
pip install --upgrade gdown 
gdown https://drive.google.com/u/0/uc?id=1jnUb-EvBw9DcwyWUQyWDdN9o42BPH7uT

実行

python run.py --data_root images --mask_root masks --model_path checkpoint_paris.pth --test --mask_mode 2

mask_mode 1　にするとランダムマスクを生成して適用してくれる。
その場合、model.pyの87行目をコメントアウトする必要がある。

            # masks = torch.cat([masks]*3, dim = 1)

🐣

---

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

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

Twitter
Medium
GitHub