摘要
本发明提供一种基于图像的轻量化车辆识别方法,包括接受车辆图像;标准化车辆图像的尺寸与通道;利用经训练的深度学习模型对标准化的车辆图像内所有车辆的边界框和类别进行标注;将标注完成的车辆图像转化为原图像大小后进行输出。本发明的有益效果是提出的轻量化车辆识别方法使架构参数量显著减少,可以降低运行模型的硬件要求,使其更易于迁移到各类造价更低、性能较差的边缘设备实时运行;通过使用的多标签损失函数可以解决图像中车辆类别不均匀的问题,提高车辆识别的准确率,尤其在少数类车辆的漏检率减少。
技术关键词
车辆识别方法
深度学习模型
图像
全局平均池化
数据标签
ReLU函数
代表
通道
模块
注意力
多标签
元素
参数
训练集
样本
编码
尺寸