🦙 LaMa: Resolution-robust Large Mask Inpainting with Fourier Convolutions
使用官方的安装方法有几个坑:
conda_env.yml
中,所有依赖均和python3.6绑定,而py36在新版的vscode中无法进行调试。(不过可以通过将python扩展进行降级的手段解决此问题,但感觉还是缓兵之计,而且,插件栏总提示要更新,强迫症不能忍 😦)- 官方使用的是
cudatoolkit=10.2
,无法与我的三张4090兼容conda_env.yml
中的好多安装包是安装其他模块时附带安装的,无需全部按照他给定的版本进行安装。此外,按照他给的yml文件进行安装,在升级某个模块时会报出依赖冲突问题。omegaconf
版本升级后,需要改动很多地方,例如:
weights_path: ${env:TORCH_HOME}
需要改为weights_path: ${oc.env:TORCH_HOME}
以及👇
`# @package _group_` 需要改为
defaults:
- _self_
- 安装推荐: 先使用conda建立一个python=3.7的环境(使得vscode可以debug),接着,安装
pytorch1.10.1
以及对应版本的pytorch_lightning,然后,安装Detectron2,最后,使用pip install -r requirements.txt
安装剩下的依赖。
pytorch1.10.1
安装指令:
# CUDA 10.2
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch
# CUDA 11.3
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
# CPU Only
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cpuonly -c pytorch
launch.json
文件的编写:
{
"version": "0.2.0",
"configurations": [
{
"name": "lama", #给调试文件起名字
"type": "debugpy", # 调试器
"request": "launch",
"program": "${workspaceFolder}/bin/train.py", # 主文件名
"console": "integratedTerminal",
"args":