今年提出的 U^2-Net 显著性检测算法（GitHub链接），刷爆了 reddit 和 twitter，号称是 2020 年「地表最强」的静态背景分割算。我们可以在JackCui的公众号文章中看到中文版的介绍。这两天画了一点时间研究这个网络，并且在自己的win10电脑上跑通了程序，得到了AI生成肖像画，接化发浑元形意太极门马掌门镇楼，效果如下：

一、运行环境

1.1 显卡硬件准备

我们需要建立一个合适的运行环境，这个折腾了我很久。首先我们的电脑得有一张Nvidia的显卡，我用的是RTX 2070，有8GB的显存，这个可以通过nvidia-smi在cmd窗口中看到相关的信息。

1.2 cuda和cudnn的安装

然后，我们需要安装CUDA和CUDNN，注意根据最新的驱动版本，我安装的是V11.1cuda是，具体的安装教程可以参考B站视频【CUDA安装】如何在Win10下轻松安装CUDA和Cudnn 手把手教学 5min，真的是手把手每一步都写的很清楚了。

1.3 虚拟环境的搭建

这里我推荐使用conda新建一个python3.6的环境，具体的创建和激活步骤可以参考博文用conda创建python虚拟环境。我创建的虚拟环境名为py36u_2net，然后需要再该虚拟环境下安装这个程序所以来的包和库。我整理如下：

numpy == 1.15.2
scikit-image == 0.14.0
opencv-python
pillow == 5.2.0
scipy
matplotlib

我们可以把以上的内容写在requirements.txt文件中，然后使用一下指令一键安装:

pip install -r requirements.txt

1.4 pytorch的安装

这步之所以单独拎出来，是因为有坑，我在这里浪费了不少于5个小时（期间走了很多弯路，做了很多无用功），我们需要再pytorch的官网，根据自己的操作系统等信息得到安装指令。

pip install torch===1.7.0+cu110 torchvision===0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

这样，就算是配好了环境，这可以说是主要的工作了哈哈。

二、代码下载

我们可以直接从github上clone代码，然后按照github上的指示一步一步操作，上面说的比较清楚了。其实对于肖像画声生成这个小功能，我们只需要运行u2net_portrait_test.py这个脚本就可以，用到的模型权重是u2net_portrait.pth文件，把他放到./saved_models/u2net_portrait/.文件夹即可。当然，如果你嫌麻烦，也可以直接下载使用Jack提供的包体（不用修改任何代码）。下载点击这里（提取码：jack）。

三、运行代码

把你需要转换的图片（比如mbg.png）存放到./test_data/test_portrait_images/portrait_im/.目录下，然后在工程的主目录下敲下以下指令：

python u2net_portrait_test.py

下面就是见证奇迹的时候啦，你将在.\test_data\test_portrait_images\portrait_results中看到转换之后的肖像画效果，是不是很赞？！

如果觉得对你有帮助，别忘了一键三连哦o(￣︶￣)o

win10完美复现U^2-Net神经网络人物肖像画生成，街头画师都要失业了相关推荐

「完美复刻」的人物肖像画生成算法 U^2-Net
一.前言大家好,我是 Jack . 今年提出的 U^2-Net 显著性检测算法,刷爆了 reddit 和 twitter,号称是 2020 年「地表最强」的静态背景分割算法,可以看下效果: 你以为今 ...
c++重写卷积网络的前向计算过程，完美复现theano的测试结果
本人的需求是: 通过theano的cnn训练神经网络,将最终稳定的网络权值保存下来.c++实现cnn的前向计算过程,读取theano的权值,复现theano的测试结果本人最终的成果是: 1.卷积神经 ...
【已开源】mtcnn_pytorch完美复现
零 .代码发现开源QQ群:414475612,名称LazyNet,代码详见群文件! GITHUBhttps://github.com/samylee/mtcnn_pytorch 一.算法介绍 MTC ...
Apple芯片Mac安装ARM版Win10完美教程（结尾附视频）
写在前面:博主是一只经过实战开发历练后投身培训事业的"小山猪",昵称取自动画片<狮子王>中的"彭彭",总是以乐观.积极的心态对待周边的事物.本人的技 ...
win10 ST-GCN复现
win10 ST-GCN复现下载源码 https://github.com/yysijie/st-gcn 安装requirements 1.anaconda切换python环境命令为 conda ...
WIN10 完美安装运行BCB6
WIN10 完美安装运行BCB6 1.右键管理员运行安装程序,尽量不要安装在C盘安装路径不要包含中空及空格 2.安装完后快捷方式右键--兼容性选择WINXP SP2 ,并设置管理员运行 WIN10 ...
Pytorch 使用GAN实现二次元人物头像生成保姆级教程（数据集+实现代码+数学原理）
Pytorch 使用DCGAN实现二次元人物头像生成(实现代码+公式推导) GAN介绍算法主体推导证明(之后将补全完整过程) 随机梯度下降训练D,G DCGAN介绍及相关原理 Py ...
Hive环境的安装部署（完美安装）（集群内或集群外都适用）（含卸载自带mysql安装指定版本）...
Hive环境的安装部署(完美安装)(集群内或集群外都适用)(含卸载自带mysql安装指定版本) Hive 安装依赖 Hadoop 的集群,它是运行在 Hadoop 的基础上. 所以在安装 Hive 之 ...
简单人物画像_怎样把复杂的人物肖像画简单化
原标题:怎样把复杂的人物肖像画简单化人物肖像画虽然很复杂但通过我们接下来全面的分析努力的你一定可以解决这些问题炯炯有神的眼睛眼睛作为一个完整的球体,这个球体会对不同视角下眼睛的样子产生影响. ...

win10完美复现U^2-Net神经网络人物肖像画生成，街头画师都要失业了