视频去水印工具 (VSR)详细安装教程

2 篇文章 0 订阅
2 篇文章 0 订阅

文章目录

  • 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):

下载解压后,双击【启动程序.exe】即可。

在这里插入图片描述

仅供具有Nvidia显卡的用户使用(AMD的显卡不行)

2.2 源码下载

源码下载

2.3 使用conda安装

2.3.1 下载安装Miniconda

2.3.2 创建并激活虚拟环境

  1. 切换到源码所在目录:

    cd <源码所在目录>
    

    例如:如果你的源代码放在D盘的tools文件下,并且源代码的文件夹名为video-subtitle-remover,就输入

    cd D:/tools/video-subtitle-remover-main
    
  2. 创建激活conda环境

    conda create -n videoEnv python=3.8
    conda activate videoEnv
    
  3. 下次进入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
      
  • 安装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,推荐使用整合包:,如果大家对这个使用有问题的话,欢迎评论区告诉我,我再出一篇文章讲解。

打开界面
在这里插入图片描述

如果这篇文章对你有用的话,请帮忙点个关注、点赞、收藏,若有其它问题,可评论区回复,谢谢~

  • 9
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值