如何使用Watermark Removal去除图片水印？

2025-09-05

1.7 K

Complete User's Guide

clone warehouse::git clone https://github.com/zuruoke/watermark-removal
Installation of dependencies：需特定版本的TensorFlow和附加库
pip install tensorflow==1.15.0
pip install git+https://github.com/JiahuiYu/neuralgym
Download model：获取预训练模型并放入model/目录

Prepare the image：将待处理图片放入项目目录
execute a command::
python main.py --image 输入路径 --output 输出路径 --checkpoint_dir model/ --watermark_type istock
Parameter description::
- –image：指定输入图片路径
- –output：设置修复后保存路径
- –watermark_type：支持不同类型水印处理

首次使用建议在Google Colab云环境中运行，可避免本地环境配置问题。对于复杂水印，可能需要调整模型参数或重新训练。