PhotoSketch

佐證：從圖像推斷輪廓圖

此存儲庫包含我們草圖生成器的培訓和測試代碼。我們還提供了[預訓練的模型]。

有關技術詳細信息和數據集，請參閱[紙張]和[項目頁面] 。

設定

現在，該代碼已更新以使用pytorch 0.4，並在Windows，Mac和Linux上運行。有關Pytorch 0.3（僅Linux）的過時版本，請查看分支Pytorch-0.3-Obsolete。

Windows用戶應找到相應的.cmd文件，而不是下面提到的.sh文件。

單行安裝（帶有CONDA環境）

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的舊版本。