Github项目推荐 | Photo-Sketching 从图片到涂鸦：高品质涂鸦的自动生成（推断）

阅读前须知：访问划线部分链接，请点击文末【阅读原文】

本项目包含草图生成器的训练和测试代码，我们还贴心地提供了【预训练模型】。

有关技术细节和数据集，请参阅【论文】和【项目页面】。

Github项目地址：

https://github.com/mtli/PhotoSketch

论文、项目Homepage：

http://www.cs.cmu.edu/~mengtial/proj/sketch/

配置

目前它只支持PyTorch 0.3。 PyTorch 0.4的支持即将推出！

单线安装（带Conda环境）

conda env create -f environment.yml

然后激活环境（sketch），准备工作就几乎完成了！

有关conda环境的更多信息，请参见此处。

手动安装

查看 environment.yml 可以获取依赖项列表。

使用预训练模型

• 下载预训练模型

• 修改 scripts/test_pretrained.sh 中的路径

• 从repo的根目录运行 scripts/test_pretrained.sh

它支持一整个图像文件夹作为输入。

训练和测试我们的轮廓绘图数据集

• 从项目页面中下载图像和渲染的草图

• 将它们解压缩并将文件排列成以下结构：

• 修改 scripts/train.sh 和 scripts/test.sh 中的路径

• 从repo的根目录，运行 scripts/train.sh 来训练模型

• 从repo的根目录，运行 scripts/test.sh 以测试val集或测试集（由阶段标志指定）

引用

如果你使用代码或数据进行研究，请引用该论文：

@article{LIPS2019,
  title={Photo-Sketching: Inferring Contour Drawings from Images},
  author={Li, Mengtian and Lin, Zhe and M\v ech, Radom\'ir and and Yumer, Ersin and Ramanan, Deva},
  journal={WACV},
  year={2019}
}

致谢

此项目代码基于旧版的pix2pix。

【AI求职百题斩 - 每日一题】

赶紧来看看今天的题目吧！

想知道正确答案？

点击公众号菜单栏【每日一题】→【每日一题】或在公众号回复“0125”即可答题获取！

点击阅读原文，查看更多内容