一、介绍
Rembg，全称为“Remove Background”，是一款基于深度学习的图像背景去除工具。它的主要功能是通过智能识别图像中的前景物体，并将其从背景中分离出来，从而创建具有透明背景的图像。
二、基础环境
系统：Ubuntu系统，
显卡：3060，
显存：12G，cuda12.1
特别提醒：建议使用cuda11.8或低版本
1.查看系统是否有Miniconda3的虚拟环境
conda -V如果输入命令没有显示Conda版本号，则需要安装。
2.更新系统命令
输入下列命令将系统更新及系统下载
apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
3.创建虚拟Python环境
-
创建一个名为"rembg"的虚拟镜像，python版本为3.8
conda create -n rembg python=3.8
-
激活“rembg"虚拟环境
conda activate rembg
4.下载模型
输入下列命令对rembg模型进行下载同时进入项目中
git clone https://gitclone.com/github.com/danielgatis/rembg.git
cd rembg/
5.下载模型依赖包
进入模型文件并下载模型依赖包
pip install -r requirements.txt
系统下载过程中，切勿有任何操作，直到出现“Successfully installed......”开头则安装结束并且成功。
6.下载缺失包
-
onnxruntime
pip intall onnxruntime
- rembg命令
pip install rembg[gpu,cli] # for library + cli
7.运行
使用rembg时列举两种方法：使用命令和web界面
7.1命令运行
rembg i /rembg/examples/animal-1.jpg /rembg/examples/animal-11.jpg
- 运行命令后会自动下载缺失的权重文件，建议单击网址后下载上传到“/root/.u2net/u2net.onnx”
- 运行结束后到“/rembg/examples”查看结果
animal-1.jpg（原图）
animal-11.jpg（结果图）
7.2Web界面
- 输入下列命令
rembg s --log_level info
- 打开网址后：
