インペインティングモデル LBAM_Pytorch

画像の一部を修復するモデル

使い方

リポジトリからPreTrainedモデルをダウンロードして入力とマスクと出力先を指定して実行します。

git clone https://github.com/Vious/LBAM_Pytorch.git
cd LBAM_Pytorch
pip install --upgrade gdown
gdown https://drive.google.com/u/0/uc?id=1blybD3ptixSmYI01lV_pWwRcO_dfYHPp

python test.py --input input.png --mask mask.png --output result.png --pretrain LBAM_NoBN_Places10Classes.pth

🐣

---

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

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

Twitter
Medium
GitHub