下载源代码
https://github.com/Sanster/IOPaint
https://github.com/advimman/lama
创建虚拟环境
conda create --prefix=D:\CondaEnvs\iopaint python=3.10
conda activate D:\CondaEnvs\iopaint
安装依赖包
pytorch最低2.0.0
# 查看cuda版本安装对应的pytorch
nvcc -V
# CUDA 11.7
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.7 -c pytorch -c nvidia
pip install iopaint
下载模型https://www.iopaint.com/install/download_model
可以走镜像https://hf-mirror.com/
这里下载lamahttps://github.com/Sanster/models/releases/tag/add_big_lama
运行
iopaint start --model=lama --device=cpu --port=8080 --model-dir=D:\iopaint
等待下载完成然后启动,其实上面也可以自己下载,注意存放路径
访问localhost:8080
至此iopaint安装成功,剩下的就是加载图片操作了,总体效果还行,密集的话就不行了