Linux下使用mmdetection的docker容器训练自己的数据
1-mmdetection介绍
mmdetection是商汤科技（2018 COCO 目标检测挑战赛冠军）和香港中文大学开源的一个基于Pytorch实现的深度学习目标检测工具箱mmdetection，支持Faster-RCNN，Mask-RCNN，Fast-RCNN，Cascade-RCNN等主流的目标检测框架。相比于Facebook开源的Detectron框架，作者声称mmdetection有三点优势：performance稍高、训练速度稍快、所需显存稍小。
github：https://github.com/open-mmlab/mmdetection
2-为什么使用docker
使用docker的优点是直接使用别人配置好的镜像来生成容器，无需配置环境，也不会有cuda版本冲突等问题。docker的安装直接使用yum install docker或者apt-get install docker来安装，为了能在容器内使用GPU，还需安装nvidia-docker，nvidia-docker的安装需要选择docker的对应版本，具体安装方法可百度。
3-标注自己的数据
安装labelme，使用pip
pip install labelme
在命令行输入labelme，启动labelme进行数据标注，如果是目标检测就用矩形框标注，是分割算法就用多边形做精细标注。标注后会得到json格式的标注文件，一张图片对应一个文件，但是这个格式的文件不是我们需要的，所以进行下一步。
4-把labelme标注的数据格式转化为coco的数据格式
coco的数据格式是指以下的数据格式，annotation中的json文件是所有图片的标注信息的集合，train和val中存放训练用和测试用的图片。格式转换代码可百度下载。
格式转换可参考这篇：https://blog.csdn.net/wc781708249/article/details/79611536
- annotations
- instances_train.json
- instances_val.json
- train
- train_1.jpg
- …
- val
- val_1.jpg
- …
更详细的数据格式可参考：https://blog.csdn.net/hajlyx/article/details/83542167
5-使用docker容器下载mmdetection并训练
输入以下命令查找镜像：
docker search mmdetection
下载这个镜像
docker pull cheney0813/mmdetection
启动镜像
nvidia-docker run -it -v /home:/mnt --shm-size 4G cheney0813/mmdetection /bin/bash
-v是挂载本地硬盘空间，–shm-size是分配共享内存
进入mmdetection根目录
cd /home/mmdetection/mmdetection
激活虚拟环境
conda activate mmdetection
创建数据链接
ln -s 你的数据路径（annotations的上一级目录） data
修改配置文件
vim configs/mask_rcnn_r101_fpn_1x.py
修改类别数：
修改标注文件路径和原始图片路径为自己的数据路径：
启动训练
./tools/dist_train.sh configs/mask_rcnn_r101_fpn_1x.py 4
大功告成！