改进YOLOv2算法的道路摩托车头盔检测

作     者：冉险生 陈卓 张禾 Ran Xiansheng;Chen Zhuo;Zhang He

作者机构：重庆交通大学机电与车辆工程学院重庆400074 

基　　金：重庆市科技局2020重庆市技术创新与应用发展专项面上项目(cstc2020jscx-msxmX0161)资助 

出 版 物：《电子测量技术》 (Electronic Measurement Technology)

年 卷 期：2021年第44卷第24期

页      码：105-115页

摘      要：针对摩托车头盔的传统检测方法准确率低、泛化能力差和目标检测网络参数量大难以在嵌入式设备运行的问题,提出改进的YOLOv2的MNXt-ECA-D-YOLOv2目标检测算法模型。首先引入Mobile Ne Xt网络替换YOLOv2原始骨干网络,其次在Mobile Ne Xt的沙漏块中引入密集连接结构同时在网络中引入有效通道注意力机制,然后在不同深度网络层应用不同的激活函数,最后在网络输出卷积层之前增加Drop Block模块。采用K-means聚类算法重新设计了自制数据集的先验框尺寸。实验结果表明,改进后的模型相比原始YOLOv2,在AP50指标上提高了3.53%,模型大小减少77.44%,检测速度提高了近4倍。通过对比实验可知,改进后的YOLOv2模型在保持较高的精度下模型更小,在CPU中的推理速度更快,具有一定的应用价值。

主 题 词：摩托车头盔检测 YOLOv2 MobileNetXt 有效通道注意力机制 激活函数 DropBlock 

学科分类：08[工学] 080203[080203] 0802[工学-机械学] 081201[081201] 0812[工学-测绘类] 

D　O　I：10.19651/j.cnki.emt.2107718

馆 藏 号：203107467...