一、下载lama-cleaner到本地
https://github.com/Sanster/lama-cleaner
二、安装miniconda
三、创建虚拟环境
conda create -n lama python=3.8
conda activate lama
进入项目根目录
先安装opencv-python
pip3 install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
安装所有依赖
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
安装显卡驱动:安装启动yolo5教程_苍穹之跃的博客-CSDN博客
安装pytorch
conda install pytorch==1.10.0 torchvision==0.11.0 cudatoolkit=11.3 -c pytorch -c conda-forge
四、启动项目
python main.py --model=lama --device=cpu --port=8080
此时系统自动会去远程下载模型(大概会失败!!!)
如果出现错误:urllib.error.URLError: <urlopen error [SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed: unable to get local issuer certificate (_ssl.c:997)>
则去main.py中加入一下代码:
import ssl
ssl._create_default_https_context = ssl._create_unverified_context
此时打开梯子,然后再重新启动即可。