国内刊号:51-1307/TP
国际刊号:1001-9081
发布日期:
作者:刘虎, 周野, 袁家斌
单位:南京航空航天大学 计算机科学与技术学院, 南京 210000
关键词:车型精细识别,卷积神经网络,双线性卷积神经网络,中心损失,多尺度
基金:江苏省产学研前瞻性联合研究项目(BY2016003-11)。
针对多角度下车辆出现一定的尺度变化和形变导致很难被准确识别的问题,提出基于多尺度双线性卷积神经网络(MS-B-CNN)的车型精细识别模型。首先,对双线性卷积神经网络(B-CNN)算法进行改进,提出MS-B-CNN算法对不同卷积层的特征进行了多尺度融合,以提高特征表达能力;此外,还采用基于中心损失函数与Softmax损失函数联合学习的策略,在Softmax损失函数基础上分别对训练集每个类别在特征空间维护一个类中心,在训练过程中新增加样本时,网络会约束样本的分类中心距离,以提高多角度情况下的车型识别的能力。实验结果显示,该车型识别模型在CompCars数据集上的正确率达到了93.63%,验证了模型在多角度情况下的准确性和鲁棒性。
来源:2019年第8期
《计算机应用》期刊编辑部