使用IOPaint实现图片擦除路人

IOPaint 是一个免费的开源的 inpainting/outpainting 工具，由最先进的 AI 模型提供支持。

IOPaint 中使用各种模型来修改图像：

• 擦除：删除任何不需要的物体、缺陷、水印、人物。
• 修复：对图像的特定部分进行修改、添加新对象或替换图片上的任何内容。
• 外画：在图像周围生成新像素以使其更大。

项目源码

Sanster/IOPaint：由 SOTA AI 模型提供支持的图像修复工具。从您的照片中删除任何不需要的物体、缺陷、人物，或擦除和替换（由稳定扩散提供动力）图片上的任何内容。 (github.com)https://github.com/Sanster/IOPaintIOPaint 提供了一个方便的 webui，使用最新的 AI 模型来编辑图像。

安装IOPaint

pip install torch==2.1.2 torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu118
pip install iopaint

启动IOPaint

iopaint start --model=lama --device=cpu --port=8080

启动后，可以通过访问 http://localhost:8080 来开始使用 IOPaint。

所有模型将在启动时自动下载。如果要更改下载目录，可以添加 --model-dir 参数。

iopaint start --model=lama --device=cuda --port=8080 --model-dir=D:\models

• --model：模型，支持lama，mat，migan，ldm，zits，fcf，manga
• --device：设备类型，支持cuda，cpu，mps
• --model-dir：模型下载路径 

批处理

iopaint run --model=lama --device=cpu --image=/path/to/image_folder --mask=/path/to/mask_folder --output=output_dir

• --image：包含输入图像的文件夹。
• --mask：包含相应蒙版图像的文件夹，所有图像都将使用此蒙版进行处理。

开发

后端

安装后端依赖

pip install -r requirements.txt

启动后端服务

python3 main.py --model lama --port 8080

 前端

安装 nodejs，然后安装前端依赖项。

git clone https://github.com/Sanster/IOPaint.git
cd IOPaint/web_app
npm install

在web_app目录下创建一个.env.local文件，并填写后端 IP 和端口。

VITE_BACKEND=http://127.0.0.1:8080

启动前端开发环境

npm run dev

然后可以访问http://localhost:5173/