代码地址
基本环境配置参考这篇文章其中,安装requirement.txt的配置的时候有个报错,把txt文件中的;sklearn换成scikit-learn在运行reqiement就行了。
pytorch和detectron我按照下面的版本不会报错(本人环境py:3.7 cuda:11.0):
conda install pytorch1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch
python -m pip install detectron2 0.3 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu110/torch1.7/index.html
配置完环境下载作者的预训练权重,加入到项目中,加入后修改脚本的几个路径为这个权重的路径。
然后修改sh脚本中gpu数量,和自己设备不符会报错。】
其余报错可参考这篇文章