经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 大数据云AI » 人工智能基础 » 查看文章
在mmdetection中跑通MaskRCNN
来源:cnblogs  作者:aimhabo  时间:2018/11/16 10:04:40  对本文有异议

1、将数据集转化成COCO格式数据集

 Kaggle->COCO:

https://github.com/pascal1129/airbus_rle_to_coco/blob/master/1_ships_to_coco.py

 

2、修改匹配的训练参数文件

/xxx/mmdetection/configs/mask_rcnn_r50_fpn_1x.py

38、52行左右的num_classes值为实际分类数+1(背景类)
100行左右的data_root值为数据集的根路径
train、val、test的ann_file、img_prefix、img_scale

中继训练时修改163行的load_from和164行的resume_from为'/xxx/xxx/epoch_xxx.pth'

 

3、训练脚本

CUDA_VISIBLE_DEVICES=0,1,2,3 python /xxx/mmdetection/tools/train.py --config /xxx/mmdetection/configs/mask_rcnn_r50_fpn_1x.py --work_dir /xxx/workspace/mask_rcnn_r50_fpn_1x --validate --gpus 3

#0,1,2,3是4张显卡全部可用的意思,具体调用几张,看最后面的--gpus 3

 

4、测试脚本

CUDA_VISIBLE_DEVICES=1,2,3 python /xxx/mmdetection/tools/test.py --config /xxx/mmdetection/configs/mask_rcnn_r50_fpn_1x.py --checkpoint /xxx/workspace/mask_rcnn_r50_fpn_1x/epoch_xxx.pth --out /xxx/workspace/out/maskrcnn.pkl --eval segm --gpus 3

 

5、图像的形式显示结果

https://github.com/open-mmlab/mmdetection/issues/92#issuecomment-438193334