1. GitHubgithub.com/zuruoke/wate
    更多AI开源软件

    使用基于机器学习的图像修复方法从图像中去除水印,这与图像的真实版本完全没有区别。

    主要功能:

    1. 水印检测:自动识别图像或视频中的水印位置。
    2. 水印去除:使用算法(如修复、插值等)移除水印,尽量还原背景。
    3. 批量处理:支持同时对多张图像或视频进行水印去除。
    4. 用户自定义:允许用户手动调整水印区域或参数,提升去除效果。
    5. 输出无水印文件:处理后的图像或视频保存为新的无水印文件。

    安装和使用

    • 使用 Google Colab
    • 首先,克隆这个仓库
      !git clone github.com/zuruoke/wate
    • 将 Directory 更改为存储库
      !cd watermark-removal
    • 由于 Google Colab 使用的是最新的 Tensorflow 2x 版本,而本项目使用的是 1.15.0,因此请降级到 Tensorflow 1.15.0 版本并重启运行时,()。although the new version of Google Colab does not need you to restart the runtime​
      !pip install tensorflow==1.15.0
    • 安装 tensorflow 工具包 neuralgym
      !pip install git+github.com/JiahuiYu/neu
    • 使用此链接下载模型目录并将其放在(重命名为,因为有时 google drive 会在下载后自动添加.txt)model/checkpoint.txtcheckpoint​

    一切就绪!!

    • 现在通过运行文件去除图像上的水印main.py​
      !python main.py –image path-to-input-image –output path-to-output-image –checkpoint_dir model/ –watermark_type istock