AI扩展程序能力:一个抠图接口的实现

156 阅读1分钟

首先,看效果

原图如下

0-拍照原图.jpg

框选抠图区间

2-鼠标框选.jpg

抠图效果

4-AI抠图后.png

项目介绍

github.com/sssxyd/remo…

Windows可执行程序

特性

  • 模型参数较少:模型大小仅44.1M,不需要高端显卡,甚至可以没有显卡(例如我的电脑)。
  • 抠图效果优异:模型的抠图效果已达到商用标准,能够满足大部分需求。
  • 代码简洁:只需少量代码即可完成模型的封装和调用。

不足

  • 复杂图片效果欠佳:对于包含多人或多物体的复杂图片,模型效果可能不理想,需要进行一定的预处理。
  • 缺乏自动补全功能:目前模型还不能自动补全被遮挡的人或物体。
  • 商用需购买License:模型briai/RMBG-1.4的商用需要购买相应的许可证。

项目背景

在传统的图像处理工作中,抠图一直是一个复杂且耗时的任务。过去,我曾尝试使用OpenCV对大量商品图片进行抠图,但始终无法解决物体轮廓的毛边问题。后来,一个朋友向我推荐了一个AI模型,briai/RMBG-1.4,它的效果让我眼前一亮。

AI的强大之处在于,它能够处理许多没有明确答案、依赖经验的任务。通过AI,我们可以将这些任务程序化,实现自动化处理,极大地提高了效率。

给我的震撼

AI从来没有离我这么近过, 它极大的扩展了程序员的能力,原来程序只能将公式、流程固化下来,现在经验也可以了! 现在开始,我们要面向AI编程了。