IOPaint是一个免费、开源且完全自托管的图像修复/扩展工具,由最先进的SOTA AI模型提供支持。
功能:
- 擦除:移除图片中不需要的对象、缺陷、水印或人物。
- 修复:对图片的特定部分进行修改,添加新对象,或替换图片中的任何内容。
- 放大:生成图片周围的新像素,图片超分辨率。
- 移除:移除背景。
并且支持使用 CPU 和 GPU,支持插件。
官方文档(包含可使用的模型以及插件):https://www.iopaint.com
GitHub地址:https://github.com/Sanster/IOPaint
注意:
使用 pip 命令需要安装 python3.x,也可以直接使用 pip3 命令,请自行安装相关环境。
擦除、去水印
可以移除图片中不需要的对象、缺陷、水印或人物
通过 pip 安装
pip install torch==2.1.2 torchvision==0.16.2
pip install iopaint
启动IOPaint
iopaint start --model=lama --model-dir=models --device=cpu --port=8080
--model=lama
指定使用lama
模型。--model-dir=models
指定存放模型的文件夹--device=cpu
指定使用CPU进行计算。--port=8080
指定服务运行在8080端口。
在启动时会自动下载模型。如果想更改下载目录,可以添加 --model-dir
参数。
如果觉得下载太慢,可以手动下载,并放在指定文件夹下面,并指定即可。
github 模型文件地址:https://github.com/Sanster/models/releases/download/add_big_lama/big-lama.pt
使用
访问本地启动的服务:http://127.0.0.1:8080 ,使用即可
照片修复
按照下面命令安装运行
pip install gfpgan
iopaint start --model=lama --model-dir=models --device=cpu --port=8080 --enable-gfpgan --gfpgan-device cpu
在运行时,会从 github 下载模型文件,同样可以手动下载放在和上面模型同样位置
github 模型文件地址:
- https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth
- https://github.com/xinntao/facexlib/releases/download/v0.1.0/detection_Resnet50_Final.pth
- https://github.com/xinntao/facexlib/releases/download/v0.2.2/parsing_parsenet.pth
图片放大、优化
按照下面命令安装运行
pip install realesrgan
iopaint start --model=lama --model-dir=models --device=cpu --port=8080 --enable-realesrgan --realesrgan-model RealESRGAN_x4plus --realesrgan-device cpu
github 模型文件地址:https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.1/RealESRNet_x4plus.pth
背景移除
按照下面命令安装运行
pip install rembg
iopaint start --model=lama --model-dir=models --device=cpu --port=8080 --enable-remove-bg
评论区