去除图像背景的工具（轻量级，批量式）

GitHub地址:danielgatis/rembg：Rembg 是一种去除图像背景的工具 (github.com)

博主在最近做一个比赛项目,然后数据用的是网上爬取的,所以数据有点脏,想清洗一下图像数据,恰好这次看到了个GitHub里面有开源去除背景的库,所以拿来用了,用了之后发现效果还是很不错的,所以在此做了个分享.

这个库的安装方法非常简单,而且用法也特别简单,跟python的os,shutil这种用法差不多,而且作者也开源了几种模型,都是通用式的,作者用了一下,命中率也达到了70以上吧,因为我要清洗的都是一些比较少见的图片,一百多M的模型能到这种效果真的很强了.

下面是他们部署的网站,可以体验一下Fast Background Remover API for developers | Photoroom

安装方法如果看不懂可以跟着我的走:

CPU版本(电脑没gpu且没CUDA):

pip install rembg[cli]

 GPU版本(电脑有GPU且安装了CUDA的):

pip install rembg[gpu,cli]

其实这样差不多已经完成了,如果后续有缺库的话可以直接对应的pip下载,因为我直接上面一键命令之后就可以运行了,通常默认cpu,听说cpu比gpu快,而且运行速度起来确实也不算很慢,大概1s两张这样.

这边是分为了命令行输入输出和导库编程输入输出两种方式,看个人选择:

 下面最重要的是选择合适的模型:

 如果一开始没有下载模型,第一次运行它会自动帮你下载默认的u2net,也就是通用模型,一般的物体去背景都用这个,下面是各种不同模型,看任务看场景选择合适的模型

所有模型都下载并保存在目录中的用户主文件夹中

最重要的一个就是不清楚用户主文件夹是什么,在哪里,一般是放在C盘用户下的,用什么模型就创建对应的.模型文件夹

使用方法大概就是这样简单高效 ,去除的效果也真的很好

 

好了,大概到这就结束了,这个库确实好用,可以用来做训练数据清洗,私人用途等等.....

 希望这篇博文对你有帮助!!!!!!