iOS上图像换背景实现

U2Net, 一个挺好的深度神经网络模型用来预估前景的范围~

rembg, 一个基于U2Net实现的python库,其中实现了一个重要算法,叫做Alpha Matting,使用这个算法可以使得半透明的部分得以预测出准确的透明度和颜色(比如头发丝的部分)

https://github.com/danielgatis/rembg/tree/main?tab=readme-ov-file

另外也可以通过机器学习的方式实现透明部分颜色的预估: https://github.com/foamliu/Deep-Image-Matting-PyTorch(这个模型太大,不适宜用,所以先不用这个)

对于系统的人像模式,系统提供了人像部分的蒙板,非常精细,详细参考WWDC:

https://developer.apple.com/videos/play/wwdc2019/260/

《iOS上图像换背景实现》有1个想法

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注