mmdetection mask-rcnn 训练步骤

mmdetection mask-rcnn 训练步骤

1、准备数据集,coco标注格式


2、修改匹配的训练参数文件
configs/mask_rcnn_r50_fpn_1x_coco.py
这里配置了模型文件,数据配置文件以及调度设置等

configs/base/datasets/coco_instance.py
修改data_root,train、val、test的ann_file、img_prefix、img_scale
configus/base/models/mask_rcnn_r50_fpn.py修改num_class为实际分类数
修改mmdet/datasets/coco.py中的CLASSES为目标的名称列表

3、训练
python tools/train.py configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.py
4、测试
python tools/test.py work_dirs/ma