近期,一家AIGC公司BRIA开源了一个出圈的模型:RMBG-1.4,它可以实现高质量地一键去除图片中的背景。下面是一些具体的例子,可以看到这个模型可以实现非常精细的“抠图”。
RMBG v1.4模型是基于22年所提出的IS-Net,但BRIA采用了私有数据集和特定的训练策略进行优化,这些改进显着提高了模型在不同图像处理场景中的准确性和有效性。
这个玩意抠图特别吊,国内很多抠图软件都要付费,公司很多设计方面的小姐姐恳求我打一套为她们服务使用,经历了一周时间,终于搭建好了,
演示地址:http://39.107.224.20:8081/
这里我记录一下搭建流程
首先需要魔法上网去这个地址 把应用down下来 https://huggingface.co/briaai/RMBG-1.4
1.1这个是官方部署的应用,这里需要用到git 将源码down下来
2.1 该应用需要使用到python3运行,我使用的服务器是centos7.8默认装有python2.5,所以我需要使用pyenv来管理多版本python
安装pyenv
图片中是百度文心给的安装示例,大家需要根据安装版本换成自己需要的python版本,本实力安装的是python 3.11.9这个版本
中途遇到了个大坑,一直安装不了,报错ssl无法导入,后来查看日志发现是 open版本需要1.1.1以上
这里参考了这边文章
https://blog.csdn.net/asdcls/article/details/129708907
python环境安装完毕后
开始安装RMBG1.4应用
进入应用目录
3.1安装需要的包
pip install -r requirements.txt
安装完成后启动应用
nohup python app.py > output.out 2>&1 &
这里会报错,应为需要去下载模型,但是国内网站无法访问
这里需要用到国内镜像 魔哈AI
https://moha.xiaoshiai.cn/models/briaai/RMBG-1.4/introduction
注册后
复制你自己的镜像加速地址
设置 hugface的站点为魔哈加速镜像地址
export HF_ENDPOINT=https://iwkudpgz.moha.xiaoshiai.cn
重新启动应用
nohup python app.py > output.out 2>&1 &
好了,成功部署,妈妈再也不用担心我抠图了