参考文档:
mmdetection入门文档
预训练权重下载:model_zoo.md
>>训练自己的模型
1.修改配置文件
位置:mmdetection/configs/__ base__/models/faster_xxxx.py
操作:
- num_class=N
- dataset_type = ‘VOCDataset’ ;data_root = ‘data/VOCdevkit/’
- ann_file = data_root + ‘VOC2007/ImageSets/Main/xx.txt’
- img_prefix = data_root + 'VOC2007/'等
说明:N表示分类类别,不包含背景,2.x版本中都不需要+1
(可选)修改输入图片大小
位置:mmdetection/configs/__base __/datasets/voc0712.py
操作:img_scale=(1000, 600)
说明:如果电脑跑不动,可以改小一点
2.1修改标签1
位置:mmdetection/mmdet/core/evaluation/class_names.py
操作:根据格式修改标签即可,以VOC为例
说明: