用AI把好朋友的照片转换为铅笔素描 —— 【🍊2020之U2Net🍊】过年回家、小学同学、初中同学、高中同学、一帮子

PK创意闹新春，我正在参加「春节创意投稿大赛」，详情请看：春节创意投稿大赛

🍊 前言

🚀 过年回家、小学同学、初中同学、高中同学、一帮子好友、不乏各种曾经的暗恋对象、就止不住的开始了双向奔赴、聚餐、high 歌、火锅嗨起来

🚀 咱也不能空着手不是、所以想了又想，决定用自己毕生所学为曾经的她们准备个小惊喜

🥇 版权: 本文由【墨理学AI】原创、各位大佬、一文读懂、敬请查阅

🎉 声明: 作为全网 AI 领域干货最多的博主之一，❤️ 不负光阴不负卿 ❤️

❤️【深度学习入门项目】❤️ 之【风格迁移】初识

🚀 本文墨理学AI 带你了解如何用AI把好朋友的照片转换为铅笔素描

❤️ 【带你了解】❤️

❤️ 人脸 ===》素描【 human portrait drawing】

💙 图像 ===》突出物体检测【Salient Object Detection】

🚀我的好朋友大概就是这样呐

❤️ 爱了爱了💙

1-3 1-4

1-5

❤️ YY 一下 💙 她看到我给她画的素描之后、很是苦恼、很后悔当初没有再勇敢一点

❤️内心暗喜 💙 我说、没关系、你现在换专业、跟我学计算机编程、一起玩AI,你也可以

2-1

2-2 2-3

🔔基础信息

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

🎄项目结构如下

🎄项目测试步骤（官方教程）

1-1

🎄项目测试（个人记录）

一：人像素描

python u2net_portrait_test.py

该命令是对整个图像进行，素描

生成效果如下：

3-1

二：人脸素描

python u2net_portrait_demo.py

会检测图像中人脸位置，只对人脸进行素描

3-2

三：测试命令

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【理解起来就是变的更好看】

❤️ 藏着掖着，不存在的, 好吧 ❤️

3-3