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