Rembg模型构建教程

一、介绍

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

• 打开网址后：