曾在字节实习的程序员小姐姐,教你一步提取动漫线稿!比用PS更清晰

每当你看到喜欢的动漫作品,想把线稿提出来、进行重新上色等加工时,你是怎么做的?

用PS吗?

是不是有点麻烦?结果还很糊?

GitHub上有一个小姐姐做了一个动漫线稿提取器,只需一行代码就能搞定!

效果是这样的:

以及这样的:

非常干净清晰有没有!

如何一步搞定动漫线稿提取

这款“又快又好”的线稿提取器名叫Anime2Sketch,可用于插画、动漫艺术和漫画。

使用Python语言开发,使用前确保具备以下条件:

  • Python 3 (推荐使用Anaconda)
  • CPU or NVIDIA GPU + CUDA CuDNN
  • Pillow, PyTorch

首先执行以下两步

  • 安装所需的软件包:pip install -r requirements.txt
  • 从GoogleDrive下载预训练权重文件,然后将其放入weights /文件夹中

然后就可以进行测试了!使用下面这行代码:

python3 test.py –dataroot 你的输入目录 –load_size 512 –output_dir 你的输出目录

其中参数load_size:由于内存限制,程序需要在处理之前调整输入图像的大小。默认情况下,将其大小调整为512×512。

这么方便,怎么实现的呢?

(小遗憾是目前只可在Linux或macOS上运行。)

基于一篇论文实现

作者介绍道,这款线稿提取器程序基于论文《用于草图到照片合成的对抗性开放领域自适应算法》。

也是由她作为一作,将在字节跳动公司实习时的一部分成果写成的论文。

论文中提到,为了从照片中合成缺少的手绘草图,他们提出了一个从草图到照片和从照

片到草图的联合学习框架。

并进一步提出了一个简单而有效的开放域采样和优化策略。

他们的方法利用所学的域内(in-domain)数据的草图到照片和照片到草图的映射,可

以迁移到开放域(open-domain)类中。

最终他们将这篇论文的研究成果应用到动漫线稿提取,做成了这样一款程序。

有网友看到后很兴奋:可以用来给自家娃做“神奇宝贝“和“爆旋陀螺”的涂色页了!

有人好奇这和用Photoshop技术上有什么不一样?其他网友解释道因为Photoshop仅检测边缘,会有“伪像”,这款明显更干净!

不过有网友指出最后的图片输出质量还可再提高,比如使用PNG格式。(原来是JPEG)

还有人说起了题外话,对示例图片里的动漫海盗战记(Vinland Saga)进行了疯狂安利!

作者介绍

向晓宇(图右),@Mukosame。普渡大学博士在读,电气与计算机工程,本科清华毕业。博士阶段研究重点是图像和视频增强,面部对齐和识别的深度学习。

今年即将加入Facebook Reality Labs实验室的On-Device AI团队,成为一名研究科学家。

感兴趣的同学戳链接尝试吧:https://github.com/Mukosame/Anime2Sketch

参考链接:

[1]https://www.arxiv-vanity.com/papers/2104.05703/
[2]https://news.ycombinator.com/item?id=27070838
[3]https://engineering.purdue.edu/people/xiaoyu.xiang.1/files/XiaoyuXiang_CV.pdf

版权所有,未经授权不得以任何形式转载及使用,违者必究。