用途:去除 Gemini 生成图片上可见的星形/闪光水印,适合清理图片、批量处理目录、发布前净化以及流水线自动化。
主要能力:基于反向 alpha 混合公式 original = (watermarked - alpha * 255) / (1 - alpha) 还原像素;内置数学生成的 alpha 图(48×48 与 96×96 两种尺寸);三阶段检测(空间 NCC 50%、梯度 NCC 30%、方差分析 20%),未检测到水印的图片自动跳过;支持 JPEG、PNG、WebP、BMP 读写,JPEG 以质量 100 输出。
使用方式:先安装依赖 pip install Pillow numpy(或 uv pip install Pillow numpy),需 Python ≥ 3.9。基本命令:python3 scripts/remove_watermark.py photo.jpg;指定输出 -o clean_photo.jpg;批量处理 python3 scripts/remove_watermark.py ./input_dir -o ./output_dir;强制处理 --force。可用参数包括 --threshold(默认 0.35)、--force-small、--force-large、--alpha-map、--verbose、--quiet。自定义 alpha 图可由白底水印截图按 alpha(x, y) = max(R, G, B) / 255 计算并保存为灰度 PNG。
适用场景:单张或批量清理 Gemini 图片、发布前处理、自动化流程集成。
限制:仅去除可见星形水印,无法去除像素级嵌入的 SynthID 隐形水印;仅处理标准右下角固定位置;内置 alpha 图为近似值,精确结果需自备参考 alpha 图;图片过小会提示且通常表示无水印。退出码 0 表示成功,1 表示有图片失败。