完整使用流程指南
环境准备
- 克隆仓库:
git clone https://github.com/zuruoke/watermark-removal
- 安装依赖:需特定版本的TensorFlow和附加库
pip install tensorflow==1.15.0
pip install git+https://github.com/JiahuiYu/neuralgym
- 下载模型:获取预训练模型并放入model/目录
实际操作步骤
- 准备图像:将待处理图片放入项目目录
- 执行命令:
python main.py --image 输入路径 --output 输出路径 --checkpoint_dir model/ --watermark_type istock
- 参数说明:
- –image:指定输入图片路径
- –output:设置修复后保存路径
- –watermark_type:支持不同类型水印处理
注意事项
首次使用建议在Google Colab云环境中运行,可避免本地环境配置问题。对于复杂水印,可能需要调整模型参数或重新训练。
本答案来源于文章《Watermark Removal:开源去除图像水印工具,图片去水印恢复原始图像》