Mxnet中的Gluoncv提供darknet53和mobilenetv1的Yolov3,由于model_zoo中有很多写好的分类模型,因此可以快速地为Yolov3更换基础网络。首先需要下载Gluoncv源码并将其修改,然后可以在本地训练中import更改的模型。我们以Gluoncv中提供的mobilenetv3_large为示例,增加一个基于mobilenetv3_large的Yolov3,并命名成‘yolo3_mobilenetv3_large_voc’,若想替换其它基础网络,方法类似。
Gluoncv代码链接:https://github.com/dmlc/gluon-cv
其中已经有的Yolov3模型:
'yolo3_darknet53_voc',
'yolo3_darknet53_coco',
'yolo3_darknet53_custom',
'yolo3_mobilenet1_0_coco',
'yolo3_mobilenet1_0_voc',
'yolo3_mobilenet1_0_custom',
'yolo3_mobilenet0_25_coco',
'yolo3_mobilenet0_25_voc',
'yolo3_mobilenet0_25_custom'
我们需要的mobilenetv3_large模型文件存放在gluoncv/model_zoo中,和mobilenetv1以及众多分