一、介绍
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环境
conda create -n rembg python=3.8
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.下载缺失包
pip intall onnxruntime
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
animal-1.jpg(原图)
animal-11.jpg(结果图)
7.2Web界面
rembg s –log_level info

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容,请联系我们,一经查实,本站将立刻删除。
如需转载请保留出处:https://51itzy.com/kjqy/152440.html