1安装python3.8.5以上版本
2.安装依赖
yum install libffi-devel -y
3.正式安装
pip install rembg
4.测试
rembg -o output.png input.jpg
1、在命令行中使用
在命令行中可以对服务端图片、本地图片以及整个文件夹图片进行操作,如下:
(1)对远程图片进行操作:
curl-s http://input.png | rembg > output.png
(2)对本地图片进行操作:
rembg-o path/to/output.png path/to/input.png
(3)对整个文件夹里的图片进行操作:
rembg-p path/to/inputs
2、作为服务使用
(1)启动服务
rembg-server
(2)如果图片可以直接链接访问,则可以通过浏览器直接打开以下地址操作:
http: //localhost:5000?url=http://image.png
也可以使用HTML表单通过上传文件的形式进行操作:
< formaction= "http://localhost:5000"method= "post"enctype= "multipart/form-data">
< inputtype= "file"name= "file"/>
< inputtype= "submit"value= "upload"/>
form>
3、在docker中使用
在docker中使用也很方便,直接运行以下命令:
curl-s http://input.png | docker run -i -v ~/.u2net:/root/.u2net danielgatis/rembg:latest > output.png
4、作为库调用
直接在脚本里调用也很简便,先创建app.py,内容如下:
importsys
fromrembg.bgimportremove
sys.stdout.buffer.write( remove( sys.stdin.buffer.read))
再执行命令运行:
cat input.png | python app.py > out.png
项目样例使用效果:
高级使用
有些图片可能使用alpha matting模式(即使用 -a -ae 15选项)处理效果会更加完美,命令如下:
curl -s http: //input.png | rembg -a -ae 15 > output.png