监控去水印

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

点击我💕跳转Github仓库

🌄 框架

├── 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’切换成保存的文件夹🔜

[2024-06-29]🌟 通过对众多监控视频中的水印进行细致的识别与记录，我们已成功完成了10000+个水印数据的标注工作！
[2024-07-13]🌟 根据所标注的水印数据集，我们训练了一个高效识别水印的目标检测模型，旨在检测图片/视频中的水印！
[2024-07-15]🌟 我们将图片还原技术应用于目标检测所识别出的框选区域中，尽可能的还原被水印遮盖的部分！
[2024-07-22]🌟 我们对目标检测模型进行了轻量化处理，并引入了warm-up策略，进一步加速了模型的生成速度，成功实现在单个3060（12G）显卡上每秒处理4张1920x1080分辨率的图片。！
[Feature]🌟我们即将推出更为轻量化的检测模型和更快速的生成模型，这些模型将专门应用于处理视频中的水印，以提高视频内容分析的效率和质量。！

希望我们的工作对你的科研有所帮助！

声明：本项目仅供学术交流之用，严禁应用于任何非法、反动等场合。若有人违反此规定，将自行承担相应后果。🤯。

YiYuan lab✌️✌️✌️