文章目录
- 1、VSR 简介
- 2、安装方法
- 2.1 一键启动安装包
- 下载地址
- 2.2 源码下载
- 2.3 使用conda安装
- 2.3.1 下载安装Miniconda
- 2.3.2 创建并激活虚拟环境
- 2.3.3 安装依赖文件
- 2.3.4 运行程序
- 2.4 使用pip安装
- 2.4.1 创建并激活虚拟环境
- 2.4.2. 安装依赖文件
- 2.4.3 运行程序
- 3、使用说明
- 4、常见问题
- 4.1 提取速度慢怎么办
- 4.2 视频去除效果不好怎么办
- 4.2.1 使用STTN算法
- 4.2.2 使用LAMA算法
- 4.3 CondaHTTPError
- 4.4 7z文件解压错误
- 4.5 RTX4090使用cuda 11.7跑不起来
- 5、 去视频的移动水印和移动物体
- 如果这篇文章对你有用的话,请帮忙点个关注、点赞、收藏,若有其它问题,可评论区回复,谢谢~
1、VSR 简介
Video-subtitle-remover (VSR) 是一款基于AI技术,将视频中的硬字幕/水印(固定位置的水印)去除的软件。主要实现了以下功能:
- 无损分辨率将视频中的硬字幕/水印去除,生成去除字幕/水印后的文件
- 通过超强AI算法模型,对去除字幕文本/水印的区域进行填充(非相邻像素填充与马赛克去除)
- 支持自定义字幕/水印位置,仅去除定义位置中的字幕/水印(传入位置)
- 支持全视频自动去除所有文本(不传入位置)
- 支持多选图片批量去除水印文本
2、安装方法
2.1 一键启动安装包
下载地址
Windows GPU版本v1.1.0(GPU):
- 百度网盘: vsr_windows_gpu_v1.1.0.zip 提取码:vsr1
- Google Drive: vsr_windows_gpu_v1.1.0.zip
下载解压后,双击【启动程序.exe】即可。
仅供具有Nvidia显卡的用户使用(AMD的显卡不行)
2.2 源码下载
2.3 使用conda安装
2.3.1 下载安装Miniconda
2.3.2 创建并激活虚拟环境
-
切换到源码所在目录:
cd <源码所在目录>
例如:如果你的源代码放在D盘的tools文件下,并且源代码的文件夹名为
video-subtitle-remover
,就输入cd D:/tools/video-subtitle-remover-main
-
创建激活conda环境
conda create -n videoEnv python=3.8 conda activate videoEnv
-
下次进入conda环境,只需要在
video-subtitle-remover
目录下输入cmd
,然后输入conda activate videoEnv
就可以进入。
2.3.3 安装依赖文件
请确保你已经安装 python 3.8+,使用conda创建项目虚拟环境并激活环境 (建议创建虚拟环境运行,以免后续出现问题)
-
安装CUDA和cuDNN
- Windows用户
- Linux用户
-
安装GPU版本Paddlepaddle:
- Windows:
python -m pip install paddlepaddle-gpu==2.4.2.post117 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html
- Linux:
python -m pip install paddlepaddle-gpu==2.4.2.post117 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
- Windows:
-
安装GPU版本Pytorch:
conda install pytorch==2.0.1 torchvision==0.16.0 pytorch-cuda=11.8 -c pytorch -c nvidia
或者使用
pip install torch==2.0.1 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
-
安装其他依赖:
pip install -r requirements.txt
2.3.4 运行程序
-
运行图形化界面
python gui.py
-
运行命令行版本(CLI)
python ./backend/main.py
2.4 使用pip安装
2.4.1 创建并激活虚拟环境
(1)切换到源码所在目录:
cd <源码所在目录>
例如:如果你的源代码放在D盘的tools文件下,并且源代码的文件夹名为video-subtitle-remover,就输入:
cd D:/tools/video-subtitle-remover-main
(2)创建并激活虚拟环境:
python -m venv videoEnv
source videoEnv/bin/activate # 如果你在Linux或macOS上
videoEnv\Scripts\activate # 如果你在Windows上
2.4.2. 安装依赖文件
请确保你已经安装 Python 3.8+,使用pip创建项目虚拟环境并激活环境(建议创建虚拟环境运行,以免后续出现问题)。
安装CUDA和cuDNN
根据你的操作系统,安装适合的CUDA和cuDNN版本。
安装GPU版本Paddlepaddle:
- Windows:
pip install paddlepaddle-gpu==2.4.2.post117 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html
- Linux:
pip install paddlepaddle-gpu==2.4.2.post117 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
安装GPU版本Pytorch:
pip install torch==2.0.1 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
安装其他依赖:
pip install -r requirements.txt
2.4.3 运行程序
运行图形化界面:
python gui.py
运行命令行版本(CLI):
python ./backend/main.py
3、使用说明
1、将这个绿框,正好框 所要去掉的水印
2、点击下面的Run
4、常见问题
4.1 提取速度慢怎么办
修改 backend/config.py
中的参数,可以大幅度提高去除速度:
MODE = InpaintMode.STTN # 设置为STTN算法
STTN_SKIP_DETECTION = True # 跳过字幕检测,跳过后可能会导致要去除的字幕遗漏或者误伤不需要去除字幕的视频帧
4.2 视频去除效果不好怎么办
修改 backend/config.py
中的参数,尝试不同的去除算法,算法介绍如下:
4.2.1 使用STTN算法
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
4.2.2 使用LAMA算法
MODE = InpaintMode.LAMA # 设置为LAMA算法
LAMA_SUPER_FAST = False # 保证效果
如果对模型去字幕的效果不满意,可以查看
design
文件夹里面的训练方法,利用backend/tools/train
里面的代码进行训练,然后将训练的模型替换旧模型即可。
4.3 CondaHTTPError
将项目中的 .condarc
放在用户目录下 (C:/Users/<你的用户名>
),如果用户目录已经存在该文件则覆盖。
解决方案:参考链接
4.4 7z文件解压错误
解决方案:升级7-zip解压程序到最新版本。
4.5 RTX4090使用cuda 11.7跑不起来
解决方案:改用cuda 11.8
pip install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
5、 去视频的移动水印和移动物体
开源工具: Propainter,推荐使用整合包:,如果大家对这个使用有问题的话,欢迎评论区告诉我,我再出一篇文章讲解。
打开界面