安装MMDetection

2020-12-05 20:33:52 阅读：1780 来源： 互联网

标签：full mmcv Pytorch CUDA 版本 MMDetection detector 安装

MMDetection是一个基于Pytorch实现的深度学习和目标检测代码库,包含了Faster-RCNN,YOLO,SSD等主流的目标检测算法代码和已经训练好的模型,方便我们进行目标检测算法的研究.MMDetection的安装步骤如下:

1. 创建一个Conda环境并Activate,很简单,就不详细说了;

2. 安装Pytorch:

conda install pytorch=1.6.0 torchvision torchaudio cudatoolkit=10.1 -c pytorch

一定要注意根据CUDA版本选择适当的Pytorch版本,比如CUDA版本是10.1,可以选择Pytorch版本为1.6.0，如果CUDA版本是10.0及以下的,需要升级到至少10.1,否则MMDetection运行时会报错;

3. 使用Git下载mmcv和mmdetection,可以从Github上下载,但是速度比较慢,建议从Gitee下载:

git clone https://gitee.com/cubone/mmdetection.git
git clone https://gitee.com/cubone/mmcv.git

4. 安装mmcv或者mmcv-full,其中mmcv是CPU版本,mmcv-full是GPU版本,根据Pytorch和CUDA版本选择合适的mmcv-full版本:

pip install mmcv-full==1.1.5+torch1.6.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html

5. 检测是否安装成功,运行下面的Demo代码:

from mmdet.apis import init_detector, inference_detector

config_file = 'configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py'
device = 'cuda:0'
# init a detector
model = init_detector(config_file, device=device)
# inference the demo image
inference_detector(model, 'demo/demo.jpg')

如果没有报错,则说明安装成功!

标签：full,mmcv,Pytorch,CUDA,版本,MMDetection,detector,安装
来源： https://www.cnblogs.com/mstk/p/14090952.html

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

ICode9

安装MMDetection