ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

mmdetection_1-环境安装mmdetection2.18.0

2021-11-07 15:33:51  阅读:248  来源: 互联网

标签:mmdetection2.18 openmmlab conda https mmdetection rcnn 安装


mmdetection_1-环境安装mmdetection2.18.0

安装mmdection2.18.0版本

官方安装步骤:https://github.com/open-mmlab/mmdetection/blob/master/docs/get_started.md

环境

操作系统:ubuntu20.04
显卡:3080Ti
显卡驱动版本:470.74
cuda:11.4

安装mmdetection

  1. 创建虚拟环境

    conda create -n openmmlab python=3.7 -y conda activate openmmlab
    创建好之后激活它
    conda activate openmmlab

  2. 安装GPU版本的pytorch

    (这里我选择pytorch1.8.0)

    conda install pytorch1.8.0 torchvision0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge(可能会有点慢,离线会快点,可参考https://blog.csdn.net/qq_37541097/article/details/120951214)

  3. 安装mmdection

    a. 推荐手动安装mmcv-full

    pip install mmcv-full==1.3.14 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html

    注意这里的mmcv、cuda、pytorch的版本要选择正确,建议查看官方的说明。

    b. 安装mmdetection(建议clone)

    git clone https://github.com/open-mmlab/mmdetection.git
    cd mmdetection
    pip install -r requirements/build.txt
    pip install -v -e . # or “python setup.py develop”

测试是否安装成功

  1. 在mmdection中创建checkpoints文件夹

  2. 下载测试的权重:https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r101_caffe_fpn_1x_coco/faster_rcnn_r101_caffe_fpn_1x_coco_bbox_mAP-0.398_20200504_180057-b269e9dd.pth

    将其放入checkpoints文件夹中

  3. python demo/image_demo.py demo/demo.jpg configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py checkpoints/xxxxx(这里是下载的权重名称)

标签:mmdetection2.18,openmmlab,conda,https,mmdetection,rcnn,安装
来源: https://blog.csdn.net/Yk_0311/article/details/121192004

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

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

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

ICode9版权所有