WatermarkRemover-AI:一款开源的AI去水印工具
WatermarkRemover-ai是什么?
WatermarkRemover-AI是一款开源、实用的AI去水印工具。它利用微软 Florence-2 技术检测水印,并借助 LaMA 模型进行修复,能有效去除各类水印。支持单张图片处理和批量操作,修复效果自然,还支持多种输出格式。无需 GPU 也能顺畅运行,还提供一键安装脚本和简单易用的界面。
WatermarkRemover-AI主要功能
双模式操作:
支持处理单个图像或整个目录中的图像。
先进的水印检测:
使用 Florence-2 的开放词汇检测功能,能够准确识别水印。
无缝修复:
使用 LaMA 模型进行上下文感知的修复,自然地填补去除水印后的区域。
自定义输出:
可以配置水印检测的最大边界框大小。
可以设置水印区域的透明度。
可以强制输出为特定格式(PNG、WEBP、JPG)。
进度跟踪:
在 GUI 和 CLI 模式下均提供实时进度更新。
暗色模式支持:
GUI 会自动根据系统设置调整为暗色模式。
高效资源管理:
支持使用 CUDA 进行 GPU 加速(可选)。
WatermarkRemover-AI技术架构
Florence-2:用于水印检测,通过开放词汇对象检测识别水印。
LaMA:用于修复,通过上下文感知的方式填补去除水印后的区域。
PyQt6:用于构建用户友好的图形界面。
WatermarkRemover-AI安装方法
克隆仓库:
git clone https://github.com/D-Ogi/WatermarkRemover-AI.git cd WatermarkRemover-AI
运行安装脚本:
使用 setup.sh 脚本自动设置环境、安装依赖并启动 GUI 应用程序。
也可以通过命令行直接使用 CLI 模式。
手动激活环境:
如果需要手动激活环境,可以运行:
conda activate py312aiwatermark
WatermarkRemover-AI使用方法
图形界面(GUI):
启动 GUI 后,可以选择处理单个图像或整个目录。
可以设置输入/输出路径、配置选项(如透明度、边界框大小等)。
点击“开始”按钮即可开始处理。
命令行界面(CLI):
基本命令格式:
python remwm.py input_path output_path [options]
常用选项:
--overwrite:覆盖已存在的文件。
--transparent:将水印区域设置为透明。
--max-bbox-percent:设置水印检测的最大边界框大小(默认为 10%)。
--force-format:强制输出为特定格式(PNG、WEBP 或 JPG)。
示例命令
python remwm.py ./input_images ./output_images --overwrite --transparent --max-bbox-percent=15 --force-format=PNG
github:https://github.com/D-Ogi/WatermarkRemover-AI