mmdetection-v2.3安装配置及训练自定义数据集

系统:Ubuntu18.04
python:3.6.9

mmdetection支持非常多的目标检测模型,从经典的Faster RCNN、SSD等,到最新的DetectoRS,还有灵活多变的配置文件,使其成为了目前最流行的目标检测框架之一。项目开发也非常活跃,这对我们用户来说是好事,这意味着拥有更稳定的版本迭代。本文主要记录使用mmdetection2.3版本训练自定义数据集的关键流程,有些细节已经省略,大家酌情参考。

  1. 安装mmcv-full。mmdetection依赖于mmcv,并且要安装pytorch对应的版本,就像torchvision一样,各版本下载链接在https://openmmlab.oss-accelerate.aliyuncs.com/mmcv/dist/index.html

  2. 源码安装mmdetection。具体安装步骤为:

    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"
    

    mmdetection2.3版本中关于CUDA/C++ 的操

你可能感兴趣的:(深度学习,Python,机器学习,mmdetection,retinanet,faster,rcnn,pytorch)