ICode9

精准搜索请尝试: 精确搜索
首页 > 系统相关> 文章详细

mmdetction linux环境搭建 v2.11.0

2021-04-25 14:02:33  阅读:224  来源: 互联网

标签:mmcv mask pytorch conda v2.11 https linux mmdetction rcnn


mmdetction linux环境搭建 v2.11.0


参考的各个网站:
https://blog.csdn.net/sinat_28442665/article/details/108059954
https://blog.csdn.net/qq_41375609/article/details/94833928
https://www.freesion.com/article/89371367948/

conda虚拟环境下的pytorch 1.3.0搭建

cuda版本:10.0 (cuda版本不能用nvidia-smi查看!)
python版本:3.6.9
pytorch版本:1.3.0 py3.6_cuda10.0.130_cudnn7.6.3_0
torchvision版本:0.4.1 py36_cu100
由于无敌长城防火墙的原因,我换conda源成中科大或清华源下载torch和torchvision都不行,所以手动去网站上下载,然后conda离线安装
pytorch包下载网址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
torchvision包下载网址:https://conda.anaconda.org/pytorch/linux-64
下载对应版本完了之后,用conda命令离线安装:

conda create -n torch130 python=3.6.9
conda activate torch130
conda install --offline xxx.tar.bz2
conda install --offline yyy.tar.bz2

最后安装完了这两个之后,用conda自动安装其他依赖包:conda

conda install pytorch==1.3.0 torchvision==0.4.1 cudatoolkit=10.0 -c pytorch

完成torch 1.3.0的安装(我之前试了很久,torch1.4.0没能成功匹配mmdet)

安装cpython

conda install cython

搭建mmcv-full

有很多地方都说这样安装mmcv:

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip install .   #注意有一个点(.)

但是用上面这个方法我匹配不上最新的mmdet,会一直error,报错mmcv库not found
因此选择了离线下载mmcv-full,然后pip手动安装:
mmcv-full下载地址:https://download.openmmlab.com/mmcv/dist/index.html
根据文件名称下载到对应版本的包,然后pip install 这个包即可,正是由于这个网址里没有torch1.4.0对应的版本,所以之前要下载1.3.0

搭建mmdetection v2.11.0

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
python setup.py develop

测试能否运行

  • 下载checkpoint
mkdir checkpoints
wget -c https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmdetection/v2.0/mask_rcnn/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco_bbox_mAP-0.408__segm_mAP-0.37_20200504_163245-42aa3d00.pth \
      -O checkpoints/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco_bbox_mAP-0.408__segm_mAP-0.37_20200504_163245-42aa3d00.pth
  • 开始infer
python3.6 demo/image_demo.py demo/demo.jpg configs/mask_rcnn/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco.py checkpoints/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco_bbox_mAP-0.408__segm_mAP-0.37_20200504_163245-42aa3d00.pth 

TODO: 修改网络代码和保存图片输出

标签:mmcv,mask,pytorch,conda,v2.11,https,linux,mmdetction,rcnn
来源: https://blog.csdn.net/qq_42044886/article/details/116126397

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有