本人不定期的基于生成技术制作一些好玩有趣的算法模型,这次带来的作品是“视频擦除”方向的应用模型,它实现的功能是自动感知到视频中我们不想看见的部分(譬如图标、字幕等等)然后进行擦除。目前已开放出“体验版”模型供大家玩耍,希望能帮助到有需要的朋友。请注意,本人制作此技术是为了帮助大家提升工作效率,请您在合规、合适的场景下使用,本人对于您的使用造成的结果概不负责。
我后续会持续不断的探索和制作新的生成方向的技术内容。基于生成模型可玩的点还有很多,此项目仅展示了其中一个做落地应用的例子。本项目的模型版权所属为:www.seeprettyface.com ,未获得授权请不要用作商业用途。关于算法的细节介绍可以参阅我的研究笔记。
图标擦除模型的功能是模型自动感知到视频中图标的位置然后进行擦除,感知图标的方法为在时域上静止不动的像素块被视作图标。
动态图标擦除模型的功能是模型自动感知到视频中动态图标的位置然后进行擦除,感知动态图标的方法为在时域上闪烁出现或动态移动的固定像素块被视作动态图标,这个在制作上有一定难度所以还没有对外开放。
字幕擦除模型的功能是模型自动感知到视频中字幕的位置然后进行擦除,感知字幕的方法为具有统一样式的文字区域被视作字幕。
torch>1.0
其他的缺什么依赖就pip install xxx,需要的东西不多
下载预训练文件放在pretrained-weight文件夹里。
预训练模型下载地址:https://pan.baidu.com/s/12Kv9DkyhLE5sWEiwm59_IA 提取码:pela
input文件夹里放置视频文件和mask文件,编辑demo.py选中任务和文件位置,然后运行python demo.py。
输入样例下载地址:https://pan.baidu.com/s/1R366Zu8TGMyv5C9kXkC9Gw 提取码:x73i
1.YoutubeVOS2018数据集;
2.基于搜集的300余部高清电影制作了2,709部电影片段数据集;
下载地址:https://pan.baidu.com/s/1CIgJmFmx5iR2JfgAyjVaeg 提取码:xb7o
3.基于搜集的40余部综艺节目制作了864部综艺片段数据集;
下载地址:https://pan.baidu.com/s/1lJk6IIWlwxknAie0LlGYOg 提取码:9rd4
4.基于搜集的180,000余张图标素材制作了36,000余张图标数据集。
第1步. 针对特定任务的时域感知训练;
第2步. 融合擦除模型的微调训练。
图标擦除模型在单卡3090上训练6天;
字幕擦除模型在单卡2080Ti上训练4天;
这个项目目前还只是自己做着好玩的,开放出的模型是比较糙的训练结果(在揉合的数据集上直接训练)。所以如果您有需求的话,可以联系我做一些更定制化的模型。当然也不局限在图标和字幕的擦除上,视频擦除的衍伸任务也有很多,譬如广告擦除、敏感内容擦除、背景人擦除等等,只要有数据的话目前的技术都是可以实现的。
我的QQ:312863063。
我的邮箱:[email protected]。
本人的研究方向是生成模型的应用技术研究。生成技术解决的问题是像素的预测,也就是在一个有缺失/完全缺失的图像棋盘上进行像素的填补/预测,使填补/预测完的图像符合真实图像的规律。基于这种模式可展开的玩法有很多,除了我之前做的数字人生成、视频内容生成等,我们还可以拓展出更多并行的思路出来。
尽管目前大部分的CV落地项目都集中在感知和识别任务上,而对于重构和生成任务的研发相对较少,但这不应影响我们对于生成技术价值的判断,毕竟生成技术是相对较新的、参与人较少的研究方向。我后续将持续致力于探索生成方向的落地型算法研发,欢迎访问我的网站了解这方面最新的研究进展:www.seeprettyface.com。