0
0

More than 1 year has passed since last update.

ランキングに入る高品質モデル

input
cased7.png
mask
case7_mask.png
result
0ed06863-445a-4cce-8ee9-38285f7344f5.png

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関連の情報を発信しています。

Twitter
Medium
GitHub

0
0
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
0
0