[python]利用神经网络算法去除照片中的背景
·
在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训练出来的多种模型:
- 通用模型(u2net)
- 专用于人物识别的模型(u2net_human_seg)。
- 轻量级模型(u2netp)
U-2-Net提供了模型的训练代码,可以自己去训练更好的模型。
3种模型各有优势,可以搭配另一种算法( PixelLib, https://github.com/ayoolaolafenwa/PixelLib)来得到更好的效果。PixelLib 可以判断图片中是否只有人物,或者有没有动物,如果没有动物,那么通过u2net_human_seg可以获取非常好的结果。如果有动物,那么通过u2net获取到的结果要比较好。

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


所有评论(0)