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盘用户下的,用什么模型就创建对应的.模型文件夹
使用方法大概就是这样简单高效 ,去除的效果也真的很好
好了,大概到这就结束了,这个库确实好用,可以用来做训练数据清洗,私人用途等等.....
希望这篇博文对你有帮助!!!!!!
