IOPaint AI 去水印实战:标注、跑批,三步出干净图
【免费下载链接】IOPaintImage inpainting tool powered by SOTA AI Model. Remove any unwanted object, defect, people from your pictures or erase and replace(powered by stable diffusion) any thing on your pictures.项目地址: https://gitcode.com/GitHub_Trending/io/IOPaint
手里一张带水印的图,后面还排着 200 张等着处理?IOPaint 是一款开源 AI 图像修复工具,核心功能是 AI 去水印和物体擦除:上传图片,把水印区域涂成蒙版,模型会用周围纹理自动补齐那块区域。全程就三步,单张大约 45 秒。这篇把选模型、安装、批量处理一次讲完。
三步出图,让水印干净消失 🎯
官方 Web 界面是 FastAPI 后端加 Web UI,流程固定为三步:
- 上传原图:常见图片格式都能直接拖进去;
- 标注水印区域:用画笔或套索工具涂出白色蒙版盖住水印。蒙版决定修复范围,宁可比水印大一圈,别抠得太紧;
- 点生成:选好修复模型,几秒后出结果。一张 1500×1004 的照片,CPU 上也是秒级返回。
下面这组是典型的水印去除样例:左图带大块版权水印,右图是 AI 修复后的成品,水印区域被背景完全补齐,放大看也找不到涂抹痕迹。
模型选不对,上面三步白做。先看怎么选。
按水印类型选模型,修复质量翻倍
IOPaint 的修复模型分两类:擦除类(自然补齐,去水印主力)和扩散类(提示词引导重绘,适合"替换"内容)。按场景对号入座:
| 模型 | 适合的水印场景 | 特点 |
|---|---|---|
| LAMA | 大面积连续水印、logo、文字 | 默认模型,CPU 上就快,效果最稳 |
| ZITS | 复杂纹理背景、半透明水印 | 边缘处理更好,保纹理 |
| Manga | 漫画网点、黑白线稿 | 专为黑白内容训练,能分清文字和线条 |
| PowerPaint / SD 修复 | 想把水印区换成别的内容 | 提示词引导,语义级修复 |
擦除类里还有 ldm、mat、fcf、migan、cv2 等备选,--model参数随时切换。纯去水印我推荐先用 LAMA,边缘不满意再换 ZITS。
带大片标题文字的海报同理:标注后修复,文字区被背景自然延伸,画面只剩视觉主体。
模型定了,剩下的就是跑起来。
安装与首次启动,两条命令进界面
跨平台最短路径,装 + 启各一行(GPU 先装 CUDA 版 PyTorch,再把--device换成 cuda):
pip3 install iopaint iopaint start --model=lama --device=cpu --port=8080跑完浏览器打开 http://localhost:8080 就是完整界面;首次启动会自动下载模型并缓存,第二次启动秒进。Windows 用户可以直接跑仓库里scripts/user_scripts/win_setup.bat这个一键脚本,省去手动装 Python 的步骤。
批量去水印,一条命令跑完全目录
200 张图别在界面里一张张涂,用run子命令。目录约定:原图放--image目录,蒙版图放--mask目录且与原图同名(黑底白区,白色即修复区,会自动缩放到原图尺寸);--mask给单个文件时,该蒙版会套用到目录里所有图片。
iopaint run --model=lama --device=cpu \ --image=input/ \ --mask=masks/ \ --output=output/跑完 output/ 里得到与原图同名的修复图,一张不落。想快速抽检质量,加--concat参数,原图、蒙版、结果会横向拼成一张对比图。
进阶玩法,三个插件补齐修复与标注
- 交互式分割:启动时加
--enable-interactive-seg,点几下就生成精准蒙版,专治毛发、边缘这类难标注区域; - GFPGAN:人脸修复,水印盖住脸的人像用得上,见
iopaint/plugins/gfpgan_plugin.py; - RealESRGAN:先 4 倍超分再修复,低清图效果更干净,见
iopaint/plugins/realesrgan.py。
三个 AI 去水印常见问题速答
CPU 跑不动或者太慢
LAMA 本身在 CPU 上就够快,512×512 的图只要几秒。卡在扩散类模型(PowerPaint 等)时,加--low-mem和--cpu-offload压显存和内存占用,或者直接换 GPU 机器跑。
修复后水印边缘有残留
多半是蒙版没盖住水印的半透明边缘。把蒙版整体加粗一圈,把渐变过渡区一起罩住再跑一次;还不满意就换 ZITS,它是纹理方向优化的模型。
首次启动很慢
首次启动慢在下载模型,不在修复。模型会缓存到本地,之后--model-dir还能指定缓存目录换位置,已下载的模型直接复用,第二次启动不再等待。
下一步可以试的:修复之外,IOPaint 同样支持 outpainting 扩展画面,给裁掉主体的图片重新"长"回完整背景,只需换模型参数。
【免费下载链接】IOPaintImage inpainting tool powered by SOTA AI Model. Remove any unwanted object, defect, people from your pictures or erase and replace(powered by stable diffusion) any thing on your pictures.项目地址: https://gitcode.com/GitHub_Trending/io/IOPaint
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考