用于一键启动video-subtitle-remover,体验Ai自动给图片视频去除水印。点击右上角“video-subtitle-remover”,等待几分钟即可出现打开的方式,直接选择通过webIDE打开即可,即可在浏览器中体验。


因为部署在服务器的docker里面,我们没有办法使用gui的界面,所以直接使用下面的命令行的工具。上传自己的文件到项目目录input下面,生成的结果也在这里面。
cd /video-subtitle-remover/ python3 ./backend/main.py
- 运行脚本后提示输入
- Please input video or image file path:
修改backend/config.py中的参数,可以大幅度提高去除速度
MODE = InpaintMode.STTN # 设置为STTN算法
STTN_SKIP_DETECTION = True # 跳过字幕检测,跳过后可能会导致要去除的字幕遗漏或者误伤不需要去除字幕的视频帧
修改backend/config.py中的参数,尝试不同的去除算法,算法介绍
InpaintMode.STTN 算法:对于真人视频效果较好,速度快,可以跳过字幕检测
InpaintMode.LAMA 算法:对于图片效果最好,对动画类视频效果好,速度一般,不可以跳过字幕检测
InpaintMode.PROPAINTER 算法: 需要消耗大量显存,速度较慢,对运动非常剧烈的视频效果较好
MODE = InpaintMode.STTN # 设置为STTN算法 # 相邻帧数, 调大会增加显存占用,效果变好 STTN_NEIGHBOR_STRIDE = 10 # 参考帧长度, 调大会增加显存占用,效果变好 STTN_REFERENCE_LENGTH = 10 # 设置STTN算法最大同时处理的帧数量,设置越大速度越慢,但效果越好 # 要保证STTN_MAX_LOAD_NUM大于STTN_NEIGHBOR_STRIDE和STTN_REFERENCE_LENGTH STTN_MAX_LOAD_NUM = 30
MODE = InpaintMode.LAMA # 设置为STTN算法 LAMA_SUPER_FAST = False # 保证效果
如果对模型去字幕的效果不满意,可以查看design文件夹里面的训练方法,利用backend/tools/train里面的代码进行训练,然后将训练的模型替换旧模型即可