Window、Linux环境下使用rembg库应用分离图片前后景

Windows环境
conda环境下,要求python版本是3.8(测试过3.6不行)
安装rembg库,pip install rembg
在命令行窗口输入rembg -p F:/input F:/output (input,output分别为输入,输出文件夹,可批量处理)
在pycharm上运行的话,导入import rembg
输入os.system(‘rembg -p F:/input F:/output’)
四张图片处理成功
在第一次运行时,若运行不出来,说明没有自动加载下载模型。需要下载模型。
https://docs.google.com/uc?export=download&id=1ao1ovG1Qtx4b7EoskHXmi2E9rp5CHLcZ
在这里插入图片描述

Linux环境下:
python3.8环境下
pip install rembg
将下载的u2net.pth 保存到根目录自创的.u2net文件夹下
运行rembg -p input output 命令即可

如报错no module named ‘torch._c’ 则安装conda install pytorch即可

#参考其他内容
https://www.jianshu.com/p/5950c40bb228

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值