作者:Xuebin Qin, Zichen Zhang, Chenyang Huang, Masood Dehghan, Osmar R.Zaiane, MartinJagersand
团队:University of Alberta, Edmonton, Canada
Github - U-2-Net
U2NET其实就是将一个unet作为基本单元,将这些基本单元以unet形式组合。
其中每一个小的单元
这里我用这个网络模拟一下人物肖像,这里主要是手绘风格,如下图:
其实说白了也是一种分割结果,无非就是一些特征分割出来,最终看起来像是人物肖像。按照这个思路其实u2net就很适合做这件事了。
首先,准备数据集,这里主要用的就是APDrawingD