WatermarkRemover

WatermarkRemover

一个基于LAMA模型的视频水印移除工具，能够批量清除视频中的固定水印。

效果展示

原始帧

去除水印

系统要求

• Python 3.10

安装步骤

• 克隆仓库

git clone https://github.com/lxulxu/WatermarkRemover.git
cd WatermarkRemover

• 创建并激活虚拟环境（可选，推荐）

python -m venv venv
# Windows
venv\Scripts\activate

• 安装基础依赖

pip install -r requirements.txt

• 安装PyTorch（二选一）

  1. CPU版本

  pip install torch
  

  1. GPU版本（需要NVIDIA显卡）

     • 安装CUDA Toolkit

       访问 NVIDIA CUDA下载页面，选择对应的操作系统和版本。

     • 安装cuDNN

       访问 NVIDIA cuDNN下载页面，选择与CUDA版本匹配的cuDNN。

     • 安装GPU版本的PyTorch

       访问 PyTorch官方网站，选择与CUDA版本匹配的命令安装，例如：

        pip3 install torch==2.6.0+cu126 torchvision==0.21.0 torchaudio==2.6.0+cu126 --index-url https://download.pytorch.org/whl/cu126
       

​ 程序会自动检测是否有可用的GPU输出相关信息并自动选择处理方式。

使用方法

基本用法

处理单个视频目录中的所有视频：

python watermark_remover.py --input /path/to/videos --output /path/to/output

带预览的处理

python watermark_remover.py --input /path/to/videos --output /path/to/output --preview

命令行参数

参数	简写	说明	默认值
--input	-i	包含视频文件的输入目录	. (当前目录)
--output	-o	处理后视频的输出目录	output
--preview	-p	启用处理效果预览	禁用

工作流程

1. 水印区域选择：程序会显示视频一帧，手动框选水印区域后按SPACE或ENTER键继续。
2. 效果预览（可选）：显示处理效果预览，按SPACE或ENTER键确认或按ESC键取消退出程序。
3. 视频处理：初次运行程序使用LAMA模型需较长时间下载模型。
4. 输出结果： MP4格式视频

局限性

• 只能处理固定位置的水印（不支持移动水印）
• 同一批处理的视频尺寸必须一致
• 同一批处理的视频水印必须一致

常见问题

Q: GPU未正确启动，程序使用CPU运行

运行时显示类似信息 No GPU detected, using CPU for processing

A: 请按照 LaMa Cleaner官方安装指南检查你的环境配置

• 检查Python版本是否为3.10

• 检查已安装PyTorch版本是否有CPU版本，参考LaMa Cleaner官方网页说明

  If Lama Cleaner is not using GPU device, it might CPU version of pytorch is installed, please follow pytorch's get-started(opens in a new tab) to install GPU version.

• 确保安装的CUDA、cuDNN和PyTorch版本和显卡兼容

  感谢@VitorX在#issue11中提供的安装步骤

  • 查询NVIDIA CUDA兼容性页面选择对应的CUDA版本
  • 查询cuDNN官方页面选择对应的cuDNN版本
  • 查询PyTorch官方页面选择对应PyTorch版本

程序正确检测到GPU会输出GPU detected: NVIDIA XXX Using GPU for processing 提示信息

Star History