AI一键抠图

逍遥一仙

前言

由于需要批量抠图，原本是想用MODNet的，可惜最新的模型官方不开源，旧模型扣的人物边缘有白边。最后发现了PP飞桨。

飞桨（PaddlePaddle）以百度多年的深度学习技术研究和业务应用为基础，集深度学习核心训练和推理框架、基础模型库、端到端开发套件、丰富的工具组件于一体，是中国首个自主研发、功能丰富、开源开放的产业级深度学习平台。

本程序封装自PP飞桨的PaddleSeg（飞桨高性能图像分割开发套件，端到端完成从训练到部署的全流程图像分割应用。）
工具内置了两个模型PP-Matting和PP-MattingV2

以下说明节选自官方文档：

PP-Matting是PaddleSeg自研的高精度抠图模型，通过引导流设计实现语义引导下高分辨率图像抠图。追求更高精度，推荐使用该模型。 且该模型提供了512和1024两个分辨率级别的预训练模型。

PP-MattingV2是PaddleSeg自研的轻量级抠图SOTA模型，通过双层金字塔池化及空间注意力提取高级语义信息，并利用多级特征融合机制兼顾语义和细节的预测。 对比MODNet模型推理速度提升44.6%， 误差平均相对减小17.91%。追求更高速度，推荐使用该模型。

图片预览

下载链接

链接：https://pan.baidu.com/s/1FCoO_dayxNOo--dTrncEUg?pwd=gprf
提取码：gprf

使用说明

1、工具应解压到非中文路径，含中文的路径必出错。
2、完整解压后，运行main.exe。将单个/多个图片拖入工具，等待提示完成后，在运行目录下的out目录内可找到扣好的图片。
3、如果需要alpha图，可勾选输出-保留蒙版

注：
1、模型菜单内，效率优先为PP-MattingV2（默认），精度优先为PP-Matting，可根据具体需求选择。
2、如果使用精度优先，等待时间会比较长
3、工具使用CPU版的PaddlePaddle，非GPU版本

逍遥一仙

前排提示：
1、此模型为人像抠图模型，不建议用来扣物。
2、中文路径下必报错，并不是说英文路径下一定不报错哈~

目前已发现的报错提示有：
1、ValueError: all the input array dimensions for the concatenation axis must match exactly, but along dimension 0, the array at index 0 has size 1 and the array at index 1 has size
目前无解

2、UnicodeEncodeError: 'gbk' codec can't encode character '\xe6' in position 63: illegal multibyte sequence
使用了中文目录

3、ImportError: DLL load failed: 动态链接库(DLL)初始化例程失败。 或  ImportError: DLL load failed with error code 3221225501
可能与CPU不支持某些指令集有关，也可能和VC运行库不全有关。具体原因未知

4、OSError: Invalid enum backend type `64`.
参见：https://www.52pojie.cn/forum.php ... 789736&pid=46818075

5、 'xxxx' 不是内部或外部命令，也不是可运行的程序或批处理文件。
原本设计是会将路径转换为短路径的，可能系统里没有相关API？把程序换到路径没有空格的地方再试

梦想家C

加个123不限速网盘
https://www.123pan.com/s/vhxtVv-BEM3A.html提取码:6666

WLK970606

显示：运行结束，但出现错误，请将log文件发送给作者 这个是什么情况

supernox

我这为啥提示运行结束，但出现错误，请将LOG文件发送给作者？？？

逍遥一仙

WLK970606 发表于 2023-5-25 17:19
额，我看到了，如下

[2023-05-25 17:08:02,295] [信息] 2023-05-25 17:08:01        

启动时没提示你不要放中文路径吗

qinxing

刚下了，拖入图片后提示 运行结束，但出现错误
ImportError: DLL load failed: 动态链接库(DLL)初始化例程失败。

itanium

逍遥一仙 发表于 2023-5-25 16:59
模型都已经放本地了

不错，可以用，已经能正常生成图片。
缺点：图片有些细小的地方也会给去掉
优点：边缘处理的还行

逍遥一仙

JakerPower 发表于 2023-5-26 04:01
报错：
[8364] Failed to execute script 'predict' due to unhandled exception!

问题：报错提示为OSError: Invalid enum backend type `64`.


查询结果：paddle从2.3开始在phi模块里用到了__lzcnt64指令 这个指令只有在支持BMI的CPU上才有，不支持BMI的CPU会跑出错误的结果。
不过该问题于两周前已在develop版本中修复，待正式实装后再重新打包哈

参见：https://github.com/PaddlePaddle/Paddle/issues/49133

逍遥一仙

jiujiu365 发表于 2023-5-27 11:11
为什么文件拖不进去，拖到窗口是显示禁止标志

不要管理员身份运行

15935517293

这个只能扣人物吗？

月影残

沙发，下载了

沐雪寒

非常感谢楼主的分享!支持

liuhn

感觉不错，早都相有一个这样的软件用了 感谢分享，支持

摇裤儿

看起来好厉害

xhsj1997

学习了，感谢

clewis

以前有几个类似软件在用，先试试看看。

dabingda

感谢楼主，谢谢啦！

BeauMxp

感谢分享