前言
???? 过年回家、小学同学、初中同学、高中同学、一帮子好友、不乏各种曾经的暗恋对象、就止不住的开始了双向奔赴、聚餐、high 歌、火锅 嗨起来
???? 咱也不能空着手不是、所以想了又想，决定用自己毕生所学为曾经的她们准备个小惊喜
- 版权: 本文由【墨理学AI】原创、各位大佬、一文读懂、敬请查阅
- 声明: 作为全网 AI 领域 干货最多的博主之一，❤️ 不负光阴不负卿 ❤️
❤️【深度学习入门项目】❤️ 之 【风格迁移】 初识
???? 本文 墨理学AI 带你了解如何 用AI把好朋友的照片转换为铅笔素描
❤️ 【带你了解】❤️
- ❤️ 人脸 ===》素描【 human portrait drawing】
- ???? 图像 ===》突出物体检测 【Salient Object Detection】
????我的好朋友大概就是这样呐
❤️ 爱了 爱了????
❤️ YY 一下 ???? 她看到我给她画的素描之后、很是苦恼、很后悔当初没有再勇敢一点
❤️内心暗喜 ???? 我说、没关系、你现在换专业、跟我学计算机编程、一起玩AI,你也可以
????基础信息
- U2-Net: Going Deeper with Nested U-Structure for Salient Object Detection
- arxiv.org/abs/2005.09…
- github.com/NathanUA/U-…
本博文，测试该代码的人像 - 素描 转换（风格迁移），记录环境搭建和效果测试；
????环境搭建
官方的库版本需求要求如下：
Python 3.6
numpy 1.15.2
scikit-image 0.14.0
python-opencv PIL 5.2.0
PyTorch 0.4.0
torchvision 0.2.1
glob
我自己的 Conda 环境搭建过程如下（安装torch==1.1.0 可用）：
- 服务器：ubuntu1~18.04
- Quadro RTX 5000 16G
- CUDA版本 V10.0.130
conda create -n nice python=3.6.9
conda activate nice
pip install torch==1.1.0
pip install torchvision==0.3.0
或者（建议使用 conda 来安装）
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch
pip install pillow==5.2.0
pip install opencv-python
pip install scipy
pip install thop
????项目结构如下
????项目测试步骤（官方教程）
????项目测试（个人记录）
一： 人像素描
python u2net_portrait_test.py
- 该命令是对整个图像进行，素描
- 生成效果如下：
二： 人脸素描
python u2net_portrait_demo.py
- 会检测 图像中人脸位置，只对人脸进行素描
三： 测试命令
python u2net_portrait_composite.py -s 20 -a 0.5
- -s indicates the sigma of gaussian function for blurring the orignal image and -a denotes the alpha weights of the orignal image when fusing them【理解起来就是变的更好看】
- ❤️ 藏着掖着，不存在的, 好吧 ❤️
- ❤️ 雨露均沾，大叔跑不了 ❤️
- ❤️ 原图越高清， 转换效果越惊艳 ❤️
GPU内存占用：
- 测试运行：GPU单卡最高占用 5073MiB
????总结
- 该论文本身是一篇关于 salient object detection（突出物检测） 的文章
- 不曾想有大佬 把它 用于 human portrait drawing（人物肖像素描）
- 取得不错响应，成为一个如此有趣的项目，不明觉厉...
【突出物检测】的运行效果如下
不同于图像分割，【突出物检测】只检测图像中 最可爱的 你
????可运行代码 + 数据
- ???? 博文此次运行的完整代码、博主这边正在快马加鞭进行整理
- ???? 各位小伙伴、如有需要、点赞、评论留言、晚上下班、即可安排
- ???? 快快学会这个代码、趁着春节、遇到曾经喜欢的她、大胆向前冲
???? 2021-1-11 素描数据集补充
人脸照片和对应艺术肖像画数据集 下载链接：
- cg.cs.tsinghua.edu.cn/people/~Yon…
墨理学AI
- ???? 作为全网 AI 领域 干货最多的博主之一，❤️ 不负光阴不负卿 ❤️
- ❤️ 如果文章对你有帮助、点赞、评论鼓励博主的每一分认真创作
