查看: 3877|回复: 0

AI一键『视频抠图』工具(免费|精细抠图|离线|无需额外配...

[复制链接]

48

主题

4

回帖

211

积分

高级会员

Rank: 2

积分
211
发表于 2022-4-29 20:09:26 | 显示全部楼层 |阅读模式
本人之前相对视频做一个抠图处理,奈何手工逐帧抠肯定不行,本来想在网上找找有没有好用且免费的抠图工具,看到有非常多人推荐一款在线抠图的网站,叫什么无绿幕抠图。我就抱着好奇点开试了一试,结果效果还是很差,而且只是生成了gif动图。综合观察,奈何要么收费,要么就是效果很差。
提示:工具在处理过程中还在完善。
一、Video、Picture处理效果

1.1原图

626c00252547c.png

626c0043d8803.png
2.1 原图

626c008dec834.png

3.1 原图

626c00a978dbb.png

4.1 原图

b92e2fd9be1c4b2e8c1b1fe4f42ca0aa.gif
24ac1bef7d4044cbba32cbc70f4cee9e.png
2.2 本工具处理效果

原代码为pytorch环境上训练的类Unet结构模型。主要分类编码、解码与上采样输出几个部分,具体不再赘述,可参考一下别人的分析博客
训练所使用的数据集包括:人像语义分割数据集、matting抠图数据集。采用的损失包括了:alpha通道L1损失与拉普拉斯金字塔损失、时序相关性损失、fg部分的L1损失和时序相关损失。将模型权重文件转换为onnx模型,本来想做个量化,所以现在处理时间可能稍久···,所以可以考虑后续加快预测处理速度,增加半精度选择、模型剪枝量化处理吧。

三、使用步骤
:选择所需处理的视频、图片;
Show Result:可以实时显示处理过程(但由于显示控件问题,弹出显示了),但是由于cv2显示会影响处理速度,可以考虑去掉勾勾;
mode:选择处理的是视频还是图片;
background color:选择抠图背景颜色,默认绿幕;
3.Start Run
626c00c721871.png

更新:V1.2



训练效果有限,处理速度受限于设备配置情况,后续考虑做个剪枝量化处理。
感谢大家支持!如果大家比较喜欢的话,后续可以考虑出一个视频插帧软件,让帧率从30fps直接翻几倍,顺便解决一下处理速度问题。


626c001675999.png
24ac1bef7d4044cbba32cbc70f4cee9e.png
b92e2fd9be1c4b2e8c1b1fe4f42ca0aa.gif
d879d8d6a6de4e599d184c5d6990b5fd.gif
d879d8d6a6de4e599d184c5d6990b5fd.gif
626c00c237c07.png
626c00bd20336.png
626c00b8c9ce1.png
626c00b5ad41b.png
626c00b245af5.png
626c00ae1988b.png
626c00a3c5ba3.png
物欲横流,谁主沉浮?
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表