├── 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
文件夹下,由于权重文件将进行周期性的更新...✈️ ,没有上传
安装
要求 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'切换成保存的文件夹🔜