MMdetection 训练工具使用

1、安装依赖

• Linux or macOS(Window is in experimentail support)
• Python 3.6+
• pytorch 1.3+
• cuda9.2+
• gcc 5+
• mmcv(importly)

注意：mmcv 的版本要和下载的mmdetection 版本对应哦！

 （建议在安装mmdetection之前，先把pytorch和mmcv单独安装好，mmcv直接用命令安装即可“pip install mmcv”）

1. 通过 git clone GitHub - open-mmlab/mmdetection: OpenMMLab Detection Toolbox and Benchmark 下载整个项目工具；
2. cd mmetection，然后python setup.py develop安装整个项目，时间比较长，耐心等待。

2、mmdetection具体介绍和基本使用

a)配置自己的训练数据，如图：

 

b)定义自己的数据类型

在mmdetection/mmdet/datasets 下创建检测数据类别脚本 MyData.py；

同时在同目录下的__init__.py脚本里面添加刚刚创建的自己的数据类

c) 运行

• 若是单个GPU，则：python tools/train.py $(CONFIG_FILE)
• 若是多个GPU，则：./tools/dist_train.sh $(CONFIG_FILES)$(GPU_NUM)[optional argumens]