centos7系统上实现detectron2安装并测试,以及一些注意事项

我这里只写下在centos7系统上实现安装Detectron2方法,最好在Anconda新建一个环境,防止跟其他的环境冲突,我这里新加一个detectron环境,conda create -n detectron python==3.6.5(安装官网的环境要求python>3.6,所以这里就直接安装了)。

1、环境要求

  • python>=3.6(已经通过创建环境时安装好)
  • pytorch>=1.4
  • torchvision要与pytorch版本对应

这里要说一下,安装pytorch会出现的坑,首先要查看自己系统的CUDA的版本号,然后去pytorch.org查找对应的版本

防止后面出现cuda读取错误,这个上面CUDA的最低版本是9.2,我的CUDA版本是9.0,我就只能去找以前的版本,这里我就直接给出安装方法了,最好使用pip 安装啊,带上清华源,要不然怎么安装超时

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torchvision==0.5.0

这样pytorch和torchvision安装好。

  • opencv-python
  • pycocotools
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple cython

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI
  • pillow
  • fvcore
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple 'git+https://github.com/facebookresearch/fvcore'

这个建议最好要安装一下,当然不安装运气好后面运行就不会报错,我第一次没有安装,跑demo没有报错,后来因为其他原因我重新安装居然报错误了,具体啥错误,给忘了,这里就给提个醒吧。

  • gcc和g++>=5

这个是最要命的事,本人在刚开始安装centos7的时候,自带的事gcc4.9和g++4.9的版本 ,我看了其他博客上写的是可以的,但是还是根据detectron2的install.md给的要求来吧,因此整个系统的gcc都需要升级,这个可以参考我写博客,当然网上一搜也是一大片,这里就不唠叨了。

好了,到这里要求的环境都安装好了,下面是安装detectron2了。

2、安装detectron2

其实这部分简单,根据facebook的githuab上面的来就可以了,我这里记录我使用的方法,先clone然后安装。

1.git clone https://github.com/facebookresearch/detectron2.git
2.cd detectron2
3.python setup.py build develop

到这里就OK了

3、demo测试

现在detectron2建立一个testimage文件夹,里面放几张测试图片

然后运行demo

python demo/demo.py --config-file /root/detectron2/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml --input /root/detectron2/testimage/1.jpg --output /root/detectron2/output --opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl

(我自己跑出来的结果因为其他原因就不放了啊,这里是借用了一位博主的,有些问题他总结的也很好,大家可以参考参考,我是挺感谢这位博主的,帮我解决了不少问题,要不然我还要慢慢摸索)

4、训练测试

      提醒一下,虽然demo跑成功了,但是不一定就能训练自己的模型了,我就出现这个bug了,哎,说出来都是泪,因为不能训练,大爷的我安装了几次,所以最好训练自己的数据前一定也做一次训练测试。就使用自带的小版本数据集,运行./datasets/prepare_for_tests.sh就可以了,然后执行./dev/run_instant_tests.sh,如果出现下面的结果,恭喜都安装成功了。

猜你喜欢

转载自blog.csdn.net/qq_33047753/article/details/107022768