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

穆友航2024-12-02 12:32

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

bash 复制代码

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

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

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

bash 复制代码

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

bash 复制代码

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

见文章顶部资源。

提取码：dow0

放入源码目录model下

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

这里有个代码bug：

bash 复制代码

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

需要修改为

bash 复制代码

assert image.shape == preprocessed_mask_image.shape

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