news 2026/9/6 18:14:07

IOPaint 命令行批处理指南:一条 run 命令跑通千张图片修复

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
IOPaint 命令行批处理指南:一条 run 命令跑通千张图片修复

IOPaint 命令行批处理指南:一条 run 命令跑通千张图片修复

【免费下载链接】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

IOPaint 是一个基于 SOTA AI 模型的图像修复(inpainting)工具,支持移除图片中不需要的物体、人物、水印等元素,也可通过 Stable Diffusion 擦除并替换图中任意内容。本文聚焦它的命令行批量处理能力:从环境安装到目录级批处理,帮你把"一张张手动修"变成"一条命令全搞定"。

5分钟装好环境,跑通第一张图

开始之前确认两件事:

  • 系统已安装 Python 3.8 或更高版本
  • 终端可以正常执行 pip 安装命令
git clone https://gitcode.com/GitHub_Trending/io/IOPaint cd IOPaint pip install -r requirements.txt

这三行完成仓库克隆与依赖安装。依赖清单见 requirements.txt,核心包含 torch、diffusers、opencv-python 等。安装完成后,python -m iopaint即可调用所有子命令。

最简命令:image、mask、output 三件套

批量修复的最小可运行命令只有三对路径参数:

python -m iopaint run --model lama --device cpu \ --image assets/watermark.jpg --mask assets/unwant_text.jpg \ --output results/watermark_removed.jpg

各参数职责:

  • --image:输入图片,可以是单个文件也可以是目录
  • --mask:掩码图片,标记需要修复的区域(白色=修复区,黑色=保留区)
  • --output:结果保存路径,文件或目录

执行后终端会显示带百分比的进度条,处理完成时在 output 指定位置生成一张 PNG 格式的结果图。

⚠️ 常见坑:如果--image传的是目录,--output也必须是目录,不能写成文件路径,否则程序会直接报错退出(校验逻辑在 batch_processing.py 开头几行)。

目录批量处理:命名即匹配

--image--mask都指向文件夹时,程序按文件主名(不含扩展名)自动配对,无需手动对应:

python -m iopaint run --model zits --device cuda \ --image ./photos/ --mask ./masks/ --output ./clean_photos/

匹配规则(实现在glob_images函数中):

  • 扫描目录下所有.png/.jpg/.jpeg文件
  • 以文件主名为 key 建立映射,photos/cat.jpg自动匹配masks/cat.png
  • 找不到对应掩码的图片会被跳过并记录日志,不会中断整批任务

掩码尺寸与原图不一致时,程序自动用最近邻插值缩放到原图尺寸,无需手动对齐。

三个修复模型,各有所长

--model参数决定底层用哪个网络做修复,常见选择:

模型擅长场景源码位置
lama大面积缺失快速填充,水印去除首选lama.py
zits细节纹理还原更精细,适合移除人物后保留背景zits.py
brushnet纹理保持能力强,复杂背景修复brushnet/

首次运行某个模型时,CLI 自动检测本地缓存(默认~/.cache),不存在则触发下载,无需手动操作。

用配置文件固化参数,告别重复敲命令

参数组合一旦调好,反复手敲容易出错。IOPaint 支持把参数存到 JSON 文件:

python -m iopaint dump-config > config.json

生成的config.json包含InpaintRequest的全部默认字段(步数、采样器、高清策略等),你只需修改关心的几项。之后每次运行:

python -m iopaint run --config config.json \ --image ./input/ --mask ./masks/ --output ./output/

模型名、设备、高清裁剪阈值等参数全部从配置文件读取,命令行只需给出路径,团队协作时传一份 config 就能复现结果。

显存不够?两个优化开关

处理高分辨率图片或跑扩散类模型时,显存吃紧是常态。两个常用参数:

  • --low_mem:开启 attention slicing 和 VAE tiling,以少量速度换内存
  • --cpu_offload:把模型权重卸载到 CPU 内存,显著降低显存占用

这两个开关定义在 cli.py 中,适用于启动 Web 服务或调用扩散模型时。批量run命令本身只暴露--model--device--concat等核心选项,内存调优主要服务于更重的推理场景。

两个典型场景对比

场景 A:批量去水印

原始图片(右下角有水印文字):

对应掩码(白色区域标记待修复范围):

执行命令:

python -m iopaint run --model lama --device cpu \ --image assets/watermark.jpg --mask assets/unwant_text.jpg \ --output results/watermark_removed.jpg --concat True

--concat True会把原图、掩码、修复结果横向拼成一张图,方便肉眼对比效果:

场景 B:批量移除多余人物

使用 zits 模型处理整目录后,背景纹理保留完整,人物区域被自然填充:

跑完之后,两步走向全自动化

  1. 掩码生成自动化:启用 interactive_seg.py 交互式分割插件,用 Segment Anything 模型快速圈出待修复区域,省去手动画掩码的步骤,实现"标注→处理"一条链路。
  2. 定时任务接入:把python -m iopaint run写进 crontab 或 shell 脚本监听特定文件夹,新图片落地即自动修复,搭出"丢进去就出结果"的流水线。

【免费下载链接】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),仅供参考

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/9/6 18:11:15

手机端ONNX模型推理加速完整指南:ONNX Runtime移动端部署实战

手机端ONNX模型推理加速完整指南:ONNX Runtime移动端部署实战 【免费下载链接】onnxruntime ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator 项目地址: https://gitcode.com/GitHub_Trending/on/onnxruntime 如果…

作者头像 李华
网站建设 2026/9/6 18:10:26

防触电技术考点全解析:从绝缘接地到漏电保护的安全逻辑

简介:面向电气安全课程学习与企业安全培训的PPT讲义,聚焦异常停电、异常带电、电气火灾等供电系统故障引发的人员伤害,系统梳理防触电技术的核心考点。内容分为直接接触电击预防、间接接触电击预防及其他电击预防三大部分:既涵盖绝…

作者头像 李华
网站建设 2026/9/6 18:08:36

完整产品结构设计流程详解:从3D建模到试产落地的全记录

简介:《一个完整产品的结构设计过程》是一份系统讲解消费电子产品从ID造型到结构落地全流程的PDF资料,面向结构设计工程师、ID工程师及产品开发人员。文档以实际工作步骤为主线,依次介绍ID草绘与外形图确认、MD资料核对、建模拆件、装配图设计…

作者头像 李华
网站建设 2026/9/6 18:00:55

基于AST模型的呼吸音分类:迁移学习与实战解析

简介:面向呼吸声音分类任务,一套完整的机器学习解决方案以PDF文档形式提供,基于ICBHI 2017挑战赛数据集与音频谱图变换模型(AST)实现,适合熟悉Python与深度学习基础的研究人员或工程师,尤其适用…

作者头像 李华