README_zh.md

Language

• English
• 中文

监控去水印

一种创新且高效的去水印方法，旨在精确识别并优雅去除图片中的水印

🌄 框架

├── configs
├── docs
├── modules
│   ├── models
│   │   └── detect
│   │   └── inpainting
│   └── weights
│       └── detect
│       └── inpainting
├── utils
│       └── img_tools.py
├── out
├── test_img
├── 
└── main.py

在 configs/ 文件夹下保存模型的参数

在 modules/ 文件夹下是本项目 核心代码👌

在 test_img/ 文件夹下是几张测试图片

在 out/ 文件夹下是几张测试图片

📝 简介

本项目将去除水印分成两步

• 第一步识别水印 ，主要代码在 modules/models/detect 文件夹下，权重文件保存在 modules/weights/detect文件夹下。

• 第二步去除水印 ，主要代码在 modules/models/inpainting 文件夹下，权重文件保存在 modules/weights/inpainting文件夹下，由于权重文件将进行周期性的更新...✈️，没有上传

🤗 detect权重下载（提取码：k1gv）

🤗 inpaint权重下载（提取码：7urm）

📝 使用

展示

效果演示1

效果演示2

安装

要求 python>=3.8

pip install -r requirements.txt 

运行

通过以下命令运行项目:

python main.py --input_folder '/path/to/your/input_folder' --output_folder '/path/to/your/output_folder' --config './configs/default.yaml'

• 将'/path/to/your/input_folder'切换成所需要去水印的文件夹✔️

• 将'/path/to/your/output_folder'切换成保存的文件夹🔜