【开源代码】图像水印移除-依赖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,待探索。

相关推荐
weixin_5142218515 分钟前
FDTD与matlab、python耦合
python·学习·matlab·fdtd
F_D_Z5 小时前
数据集相关类代码回顾理解 | StratifiedShuffleSplit\transforms.ToTensor\Counter
python·torchvision·transforms
tao3556676 小时前
【Python刷力扣hot100】283. Move Zeroes
开发语言·python·leetcode
小宁爱Python7 小时前
从零搭建 RAG 智能问答系统1:基于 LlamaIndex 与 Chainlit实现最简单的聊天助手
人工智能·后端·python
湖南人爱科技有限公司7 小时前
RaPhp和Python某音最新bd-ticket-guard-client-data加密算法解析(视频评论)
android·python·php·音视频·爬山算法·raphp
eqwaak08 小时前
数据预处理与可视化流水线:Pandas Profiling + Altair 实战指南
开发语言·python·信息可视化·数据挖掘·数据分析·pandas
心态特好9 小时前
详解WebSocket及其妙用
java·python·websocket·网络协议
dlraba80210 小时前
用 Python+OpenCV 实现实时文档扫描:从摄像头捕捉到透视矫正全流程
开发语言·python·opencv
小熊出擊10 小时前
【pytest】fixture 内省(Introspection)测试上下文
python·单元测试·pytest
njsgcs10 小时前
sse mcp flask 开放mcp服务到内网
后端·python·flask·sse·mcp