centos7 SOLOV2部署各个依赖库的版本
solov2 GitHub:[link]https://github.com/WXinlong/SOLO
torch 1.4.0
torchvision 0.5.0
mmcv-full 1.3.8
mmdet 1.0.0(建议按照下面的方式安装)
opencv-python 4.5.3.56(没有要求)
pycocotools 2.0.2
git clone https://github.com/open-mmlab/mmdetection.git
(如果下载不下来,可以去GitHub上先下载到本地)
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e . # or “python setup.py develop”
cuda10.0 && 10.1都行cudnn下载对应的就行~
gcc版本用到的是7,应该不能>=8
其他的按照这个博主的去训练自己的数据集和测试
.链接: [link]https://blog.csdn.net/donkey_1993/article/details/108054849.
我这边跑出来的效果