基于Dense-YOLOv3的车型检测模型

• 作者: 曹建芳 王彦苏 陈立潮
• 来源期刊: 计算机系统应用
• 关键词: 车型检测 YOLOv3 DenseNet 鲁棒性 平均精度 召回率
• 年，卷（期）: 2020,（10）
• 所属期刊栏目: 软件技术·算法
• 页码范围: 158-166
• 页数: 9页
• 语种: 中文
• DOI: 10.15888/j.cnki.csa.007634

摘要

针对传统YOLOv3的网络结构存在曝光过度或光线较暗等异常图片在提取特征时鲁棒性较差,导致车型识别率低下的问题,提出了一种用于交通车辆检测的Dense-YOLOv3模型.该模型集成了密集卷积神经网络DenseNet和YOLOv3网络的特点,加强了卷积层之间的车型特征传播和重复利用,提高了网络的抗过拟合性能;同时,对目标车辆进行了不同尺度的检测,构建了交叉损失函数,实现了车型的多目标检测.经过在BIT-Vehicle标准数据集上对模型进行训练和测试,实验结果表明,基于Dense-YOLOv3车型检测模型平均精度达到了96.57％,召回率为93.30％,表明了该模型对车辆检测的有效性和实用性.