一种改进的YOLOv5电动车头盔佩戴检测方法
An Improved YOLOv5 Electric Vehicle Helmet Wearing Detection Method
刘超 1高健1
作者信息
- 1. 江苏大学 电气信息工程学院,江苏 镇江 212013
- 折叠
摘要
利用深度学习算法检测非机动车交通违法行为有助于加快我国交通智能化发展,保障交通安全.为此,设计一种基于改进YOLOv5算法的电动车骑手头盔佩戴自动检测方法.该方法在YOLOv5算法的基础上利用Inception卷积减少特征提取网络参数量,引入注意力机制优化目标检测结果.在自建电动车头盔数据集QCKJ-MH上的实验结果表明,该方法的识别精度均值达96.4%,检测速度达82FPS,模型大小为12.9 MB,能够精准、快速地对电动车骑手头盔佩戴情况进行识别.
Abstract
The use of deep learning algorithms to detect non motorized vehicle traffic violations can help accelerate the development of intelli-gent transportation in China and ensure traffic safety.To this end,design an automatic detection method for helmet wearing of electric bike rid-ers based on the improved YOLOv5 algorithm.This method is based on the YOLOv5 algorithm and utilizes Inception convolution to reduce the parameters of the feature extraction network,introducing an attention mechanism to optimize the object detection results.The experimental re-sults on the self built electric vehicle helmet dataset QCKJ-MH show that the average recognition accuracy of this method reaches 96.4%,the detection speed reaches 82 FPS,and the model size is 12.9 MB.It can accurately and quickly identify the wearing situation of electric vehicle riders' helmets.
关键词
电动车头盔/YOLOv5/交通智能化/注意力机制/轻量化Key words
electric vehicle helmet/YOLOv5/traffic intelligence/attention mechanism/lightweight引用本文复制引用
出版年
2024
国家科技期刊平台
NSTL
万方数据