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/