近日,PramaLLC 推出了其最新的人工智能模型 —— 背景消除网络 BEN2(Background Erase Network),该模型在前景分割技术上做出了显著创新。BEN2采用了信心引导抠图(Confidence Guided Matting,CGM)管道,利用一个精炼网络专门处理基础模型信心较低的像素,从而获得更为精准、可靠的抠图效果。
BEN2的训练数据集包括 DIS5k 以及 PramaLLC 自有的22K 分割数据集。得益于这些高质量的数据,BEN2在头发抠图、4K 图像处理、物体分割以及边缘优化等方面表现出色。值得注意的是,基础模型是开源的,用户可以通过访问其官网进行尝试,也可通过 API 将 BEN2集成到自己的项目中。
对于开发者来说,安装 BEN2也十分简单。只需使用 pip 命令便可完成安装,之后便可通过几行代码快速开始图像处理工作。该模型支持单张图像及批量图像处理,批处理时推荐的最大批量为3,以确保消费者级 GPU 的最佳性能。
除了图像处理,BEN2还提供了视频分割功能。用户可以轻松将视频中的前景与背景分离,操作流程同样简单。只需指定视频路径,BEN2便能自动处理,最终生成的视频将保存为前景.webm 或前景.mp4格式,便于后续使用。
PramaLLC 在其官网提供了免费在线演示,用户可亲自体验 BEN2的强大功能。
模型:https://huggingface.co/PramaLLC/BEN2
HuggingFace在线使用:https://huggingface.co/spaces/PramaLLC/BEN2
官方:https://backgrounderase.net/home
划重点:
🌟 BEN2采用信心引导抠图技术,提升图像抠图精度。
🖼️ 支持单张及批量图像处理,操作简单易上手。
🎥 提供视频分割功能,方便用户处理视频前景和背景。