用AI把好朋友的照片转换为铅笔素描——【模型识别2020之U2Net】

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

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

---

❤️ 【带你了解】❤️

---

• ❤️ 人脸 &＃61;&＃61;&＃61;》素描【 human portrait drawing】
• &＃x1f499; 图像 &＃61;&＃61;&＃61;》突出物体检测 【Salient Object Detection】

&＃x1f680;我的好朋友大概就是这样呐

❤️ 爱了 爱了&＃x1f499;

&＃x1f514;基础信息

• U2-Net: Going Deeper with Nested U-Structure for Salient Object Detection
• https://arxiv.org/abs/2005.09007
• https://github.com/NathanUA/U-2-Net

本博文&＃xff0c;测试该代码的人像 - 素描 转换&＃xff08;风格迁移&＃xff09;&＃xff0c;记录环境搭建和效果测试&＃xff1b;

&＃x1f384;环境搭建

官方的库版本需求要求如下&＃xff1a;

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 环境搭建过程如下&＃xff08;安装torch&＃61;&＃61;1.1.0 可用&＃xff09;&＃xff1a;

• 服务器&＃xff1a;ubuntu1~18.04
• Quadro RTX 5000 16G
• CUDA版本 V10.0.130

conda create -n nice python&＃61;3.6.9conda activate nicepip install torch&＃61;&＃61;1.1.0pip install torchvision&＃61;&＃61;0.3.0
或者&＃xff08;建议使用 conda 来安装&＃xff09;
conda install pytorch&＃61;&＃61;1.1.0 torchvision&＃61;&＃61;0.3.0 cudatoolkit&＃61;10.0 -c pytorchpip install pillow&＃61;&＃61;5.2.0pip install opencv-pythonpip install scipy
pip install thop


&＃x1f384;项目结构如下

&＃x1f384;项目测试步骤&＃xff08;官方教程&＃xff09;

&＃x1f384;项目测试&＃xff08;个人记录&＃xff09;

一&＃xff1a; 人像素描

python u2net_portrait_test.py


• 该命令是瑞整个图像进行&＃xff0c;素描
• 生成效果如下&＃xff1a;

二&＃xff1a; 人脸素描

python u2net_portrait_demo.py


• 会检测 图像中人脸位置&＃xff0c;只对人脸进行素描

三&＃xff1a; 测试命令

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

---

• ❤️ 藏着掖着&＃xff0c;不存在的, 好吧 ❤️

• ❤️ 雨露均沾&＃xff0c;大叔跑不了 ❤️

---

• ❤️ 原图越高清&＃xff0c; 转换效果越惊艳 ❤️

GPU内存占用&＃xff1a;

• 测试运行&＃xff1a;GPU单卡最高占用 5073MiB

&＃x1f4ac;总结

• 该论文本身是一篇关于 salient object detection&＃xff08;突出物检测&＃xff09; 的文章&＃xff1b;
• 不曾想有大佬 把它 用于 human portrait drawing&＃xff08;人物肖像素描&＃xff09;
• 取得不错响应&＃xff0c;成为一个如此有趣的项目&＃xff0c;不明觉厉…

【突出物检测】的运行效果如下

---

不同于图像分割&＃xff0c;【突出物检测】只检测图像中 最可爱的 你

&＃x1f389;可运行代码 &＃43; 数据

考虑到一些小伙伴 官方代码 和 模型下载有困难&＃xff0c;博主这里把此次博文对应代码上传到网盘&＃xff0c;里面包含本次测试使用的代码和下载的预训练模型&＃xff1b;

---

有需要的小伙伴&＃xff0c;搜索关注博主同名公号&＃xff1a;墨理学AI &＃xff0c;公号后台&＃xff0c;回复 20200101 即可获取 云盘下载链接&＃xff1a;

20200101


&＃x1f4e2; 2021-1-11 素描数据集补充

来自 https://github.com/yiranran/APDrawingGAN 的 人脸照片和对应艺术肖像画数据集 下载链接&＃xff1a;

• https://cg.cs.tsinghua.edu.cn/people/~Yongjin/APDrawingDB.zip

&＃x1f680;&＃x1f680;文末专栏推荐部分&＃x1f680;&＃x1f680;

• &＃x1f384;如果感觉文章看完了不过瘾&＃xff0c;还想更进一步&＃xff0c;那么可以来我的其他 专栏 看一下哦~

• ❤️ 图像风格转换 —— 代码环境搭建 实战教程【关注即可阅】&＃xff01;
• &＃x1f49c; 图像修复-代码环境搭建-知识总结 实战教程 【据说还行】
• &＃x1f499; 超分重建-代码环境搭建-知识总结 解秘如何让图像变得清晰【老少皆宜】
• &＃x1f49b; YOLO专栏&＃xff0c;只有实战&＃xff0c;不讲道理 图像分类【建议收藏】&＃xff01;

• &＃x1f384; 个人整理的Cuda系列 Linux安装教程【适合小白进阶】

• &＃x1f49c; ubuntu18给当前用户安装cuda11.2 图文教程 | 配置cuDNN8.1 |
• &＃x1f49c; Linux服务器下给当前用户安装自己的CUDA10.0
• &＃x1f49c; Linux下cuda10.0安装Pytorch和Torchvision
• &＃x1f49c; Linux 可以安装多个版本的Cuda 吗 | 给我一台新的服务器&＃xff0c;我会怎么安排 Cuda
• &＃x1f49c; 查看CUDA和cuDNN的版本号