mmdetection安装教程 | 踩坑总结

mmdetection安装教程 | 踩坑总结

重磅干货，第一时间送达

导读

mmdetection是商汤科技和香港中文大学开源的一个基于PyTorch 实现的深度学习目标检测工具箱。

mmdetection 有三大优势，性能优越，训练速度快，占显存小。

重要的是，很多论文和比赛都会用这个框架去打比赛，那么对于我们直接套用他们的代码，和基于大佬们的基础上去优化更加的方便。

在国产中算是非常优秀的一个框架，当然百度的PaddleDetection也不错，最近也在白嫖百度的GPU，大致都差不多

前言

安装和配置环境最关键的是要版本对应，否则会浪费大量的时间。

1、安装MMDetection 需要安装 MMCV，所以查看这两个东西的版本要求：

master 即指目前克隆的最新 MMDetection 的版本，那么要求的 mmcv 版本要1.2.4=< mmcv <1.3

2、CUDA 、pytorch、mmcv版本选择

官方提供了一个参考图，建议按照这种方式安装，亲测比较顺利

这里以 CUDA10.1、pytorch 1.6.0、torchvision 0.7.0 做示范来安装 MMDetection

1、安装教程

官方教程：安装

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f install mmcv-full==1.2.4 -f clone mmcvMMCV_WITH_OPS=1 pip install -e . # package mmcv-full will be installed after this stepcd ..

PS：如果报错

MMCV_WITH_OPS=1 pip install -e . 报错替换成 pip install .

3、mmdetection 安装和编译

git clone mmdetectionpip install -r requirements/build.txtpip install -v -e . # or "python setup.py develop"

2、测试

测试代码: demo/inference_demo.ipynb

预训练模型：

​​文件夹下

随后运行即可！

如果你喜欢这个系列， 可以给我三连，我会努力更新下去！

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。