PhotoSketch
1.0.0

このレポは、スケッチジェネレーターのトレーニングとテストコードが含まれています。 [事前に訓練されたモデル]も提供します。
技術的な詳細とデータセットについては、 [ペーパー]と[プロジェクトページ]を参照してください。
このコードは、Pytorch 0.4を使用するように更新され、Windows、Mac、Linuxで実行されます。 Pytorch 0.3(Linuxのみ)を使用した時代遅れのバージョンについては、Branch Pytorch-0.3-Obsoleteをご覧ください。
Windowsユーザーは、以下に説明する.shファイルの代わりに、対応する.cmdファイルを見つける必要があります。
conda env create -f environment.yml
その後、環境(スケッチ)をアクティブにすると、準備ができています!
Conda環境の詳細については、こちらをご覧ください。
依存関係のリストについては、 environment.yml参照してください。
scripts/test_pretrained.shのパスを変更しますscripts/test_pretrained.shを実行します入力として画像のフォルダーをサポートします。

scripts/train.shとscripts/test.shのパスを変更しますscripts/train.shを実行してモデルをトレーニングしますscripts/test.shを実行して、VALセットまたはテストセットでテストします(フェーズフラグで指定) 調査にコードまたはデータを使用する場合は、論文を引用してください。
@article{LIPS2019,
title={Photo-Sketching: Inferring Contour Drawings from Images},
author={Li, Mengtian and Lin, Zhe and Mv ech, Radom'ir and and Yumer, Ersin and Ramanan, Deva},
journal={WACV},
year={2019}
}
このコードは、PIX2PIXの古いバージョンに基づいています。