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

下载源码

复制代码
git clone https://github.com/zuruoke/watermark-removal

创建conda环境

bash 复制代码
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环境,命令行安装

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

下载模型

谷歌云盘

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:

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,待探索。

相关推荐
快乐的钢镚子2 小时前
【leetcode hot 100】49.字母异位词分组
python·leetcode
honeysuckle_luo3 小时前
RandLA-net-pytorch 复现
人工智能·pytorch·python
SunnyDays10114 小时前
Python 高效实现 Excel 与 TXT 文本文件之间的数据转换
python·excel转txt·文本转excel·excel转文本·txt转excel
硬件人某某某5 小时前
python基于卷积神经网络的桥梁裂缝检测系统(django),附可视化界面,源码
python·cnn·django
java1234_小锋5 小时前
PyTorch2 Python深度学习 - 自动微分(Autograd)与梯度优化
开发语言·python·深度学习·pytorch2
java1234_小锋5 小时前
PyTorch2 Python深度学习 - 简介以及入门
python·深度学习·pytorch2
tao3556677 小时前
【Python刷力扣hot100】42. Trapping Rain Water
开发语言·python·leetcode
一只安7 小时前
从零开发AI(不依赖任何模型)
人工智能·python
2501_938782097 小时前
实战 Python NLP:处理 PDF 文档批量提取文本并进行主题建模
python·自然语言处理·pdf
成长痕迹7 小时前
【Python与Matlab数据分析对比】
python·matlab·数据分析