基于深度学习的智能治超场景下货车车型识别

摘要

针对智能治超场景下超载车辆自动化检测的需求，在YOLOv5s的基础上从数据、模型和算法3个方面提出了一种改进的货车车型识别算法；在数据层面，使用的数据增强模拟了现实中面对恶劣天气、图像噪声和数据损坏等复杂场景，丰富了训练数据的多样性，提高了模型在复杂场景下的鲁棒性；在模型方面，提出了一种新的注意力机制来综合考虑不同通道的重要性和编码特征的位置信息，提高了模型的识别准确性；在算法层面，针对现有算法的不足，提出了一种更通用的标准来判断货车与轮轴的隶属关系，以适用更复杂的场景；实验结果表明，提出的改进模型对货车和轮轴的识别精度分别达到99.34%和99.22%,对货车车型识别的准确率为98.71%;与经典的YOLOv5s网络相比，货车和轮轴的平均识别精度提高了2.39%,货车车型的识别准确率提高了2.22%;综上，所提出的方法实现了对货车车型自动和准确的识别，可以为智能治超场景下的货车车型识别提供理论支撑。