一种基于深度学习的精细车型识别方法及系统

基本信息

申请号 CN202110112292.8 申请日 -
公开(公告)号 CN112966709A 公开(公告)日 2021-06-15
申请公布号 CN112966709A 申请公布日 2021-06-15
分类号 G06K9/62(2006.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 汪洋旭;廖欢;张招亮;刘后标;唐文杰 申请(专利权)人 中国电子进出口有限公司
代理机构 北京君尚知识产权代理有限公司 代理人 邱晓锋
地址 100036北京市海淀区复兴路17号A座6-23层
法律状态 -

摘要

摘要 本发明涉及一种基于深度学习的精细车型识别方法及系统。该方法从车辆视频流数据中抽取视频帧,检测视频帧中的车辆并提取车辆图像;将车辆图像输入训练完成的引入注意力训练机制的深度卷积神经网络,输出精细车型识别结果。所述深度卷积神经网络包括主干特征提取网络、特征表达层、注意力特征提取分支网络、分类损失层;训练过程包括车型正常‑随机打乱图像对训练、多元组车辆图像对训练;使用的损失函数包括Softmax分类损失、多元组分类损失、分支网络重构损失。本发明基于注意力机制能够自主学习不同车型间的区分性特征,能够有效提升现有基于深度学习方法的车型识别模型的车型识别准确率,实现针对大量实际车型的精细车型识别需求。