栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/530329.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号