【开源代码】图像水印移除-依赖python-tensorflow

下载源码

git clone https://github.com/zuruoke/watermark-removal

创建conda环境

conda create -n tensorflow_gpu python=3.7
conda activate tensorflow_gpu
conda install tensorflow-gpu==1.15

安装依赖neuralgym

（试了anaconda的powershell窗口不行，报git找不到的错误）

打开git bash窗口，预先找到上述python环境，命令行安装

C:/software/anaconda3/envs/tensorflow_gpu/python.exe -m pip install git+https://github.com/JiahuiYu/neuralgym

安装其他依赖

conda install Pillow 
pip3 install opencv-python3
pip3 install tensorflow-estimator==1.15
pip3 install pyyaml

下载模型

谷歌云盘

https://drive.google.com/drive/folders/1xRV4EdjJuAfsX9pQme6XeoFznKXG0ptJ

CSDN资源

见文章顶部资源。

百度网盘

链接：https://pan.baidu.com/s/131E1PMTURjyxG5qR3EASBQ

提取码：dow0

放入源码目录model下

开始使用

下载图像，保存到本地demo_input.png

示例图像：https://user-images.githubusercontent.com/51057490/140277713-c7d6e2b9-db62-4793-823a-25ed0c4e2771.png

MASK图像

这里有个代码bug：

watermark-removal\preprocess_image.py", line 42, in preprocess_image
    assert image.shape == preprocessed_mask_image

需要修改为

assert image.shape == preprocessed_mask_image.shape

结果展示

问题

需要提前有对应的MASK图像，要么自己PS做，要么用技术生成mask，待探索。