改进Mask R-CNN的细粒度车型识别算法

作者：江昆鹏; 闫洪涛; 杨红卫; 张庆辉

来源：软件, 2020, 41(03): 1-5.

• 细粒度车型识别
• Mask R-CNN
• 聚合残差-特征金字塔网络
• 区域建议网络
• 锚
• Soft-NMS

针对车辆型号繁多、部分型号间差异较小带来车辆分类困难的问题,构建一种基于改进的Mask R-CNN细粒度车辆型号识别算法。改进后的算法采用聚合残差-特征金字塔网络(ResNeXt-FPN)提取特征图;调整了区域建议网络(RPN)中锚(Anchor)的尺寸大小;用Soft-NMS代替了非极大值抑制算法(NMS),以提高检测精度;去除掩码分支,节省了预测时间。为了验证算法改进的效果,将其与最新的目标检测算法进行对比。实验结果证明,改进的算法提高了车辆识别的准确率,比原始算法准确率提升了2%。

• 出版日期2020
• 单位河南工业大学