在github寻找了一圈,终于发现了一个比较新且好用的开源项目rembg(地址https://github.com/danielgatis/rembg)。结合rembg和PixelLib可以得到非常好的效果。

rembg 底层是u2net(https://github.com/xuebinqin/U-2-Net)—published in Pattern Recognition 2020。

U-2-Net是一个相当不多的项目,一位在加拿大阿尔伯塔大学留学的中国人做的(https://webdocs.cs.ualberta.ca/~xuebin/),可以基于它做很多关于图像处理的项目。

rembg可以选择U-2-Net训练出来的多种模型:

  1. 通用模型(u2net)
  2. 专用于人物识别的模型(u2net_human_seg)。
  3. 轻量级模型(u2netp)

U-2-Net提供了模型的训练代码,可以自己去训练更好的模型。

3种模型各有优势,可以搭配另一种算法( PixelLib, https://github.com/ayoolaolafenwa/PixelLib)来得到更好的效果。PixelLib 可以判断图片中是否只有人物,或者有没有动物,如果没有动物,那么通过u2net_human_seg可以获取非常好的结果。如果有动物,那么通过u2net获取到的结果要比较好。

在这里插入图片描述

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐