python肖像画_U2NET人物肖像画模拟

本文介绍了使用U2NET网络进行人物肖像画模拟的方法,该网络基于UNET结构,用于特征分割。通过APDrawingDB数据集训练,简单调整参数后即可开始训练。经过50个epoch的训练,得到了不错的效果,但高参数量对显存需求较大。
摘要由CSDN通过智能技术生成

作者:Xuebin Qin, Zichen Zhang, Chenyang Huang, Masood Dehghan, Osmar R.Zaiane, MartinJagersand

团队:University of Alberta, Edmonton, Canada

Github - U-2-Net

U2NET其实就是将一个unet作为基本单元,将这些基本单元以unet形式组合。

fed100b91c2151ff9ac6a45ca713e11b.png

其中每一个小的单元

3133bcf64f677e2a46563aa3409f6c95.png

这里我用这个网络模拟一下人物肖像,这里主要是手绘风格,如下图:

0ab40cad82c2eba574a70243b643021b.png

其实说白了也是一种分割结果,无非就是一些特征分割出来,最终看起来像是人物肖像。按照这个思路其实u2net就很适合做这件事了。

首先,准备数据集,这里主要用的就是APDrawingD

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值