duan2

导航

 

近期,一家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 &

好了,成功部署,妈妈再也不用担心我抠图了

posted on 2024-05-10 16:15  duan2  阅读(54)  评论(0编辑  收藏  举报