一种创新且高效的去水印方法,旨在精确识别并优雅去除图片中的水印
点击我💕跳转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
文件夹下,由于权重文件将进行周期性的更新…✈️,没有上传
🎇 使用
效果演示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✌️✌️✌️