基于改进YOLOv3算法的非机动车头盔佩戴检测方法

摘要

本发明公开一种基于改进YOLOv3算法的非机动车头盔佩戴检测方法，包括：获取交通监控视频流，进行加工、数据增强建立训练数据集；构建改进的YOLOv3目标检测模型；使用训练数据集对改进的YOLOv3算法模型进行训练，将训练后最佳的权重文件加载至模型得到非机动车头盔检测网络；读入待检测的交通监控视频帧图像，采用所述非机动车头盔检测网络输出对应的头盔检测结果。本发明解决了传统人工检测方法效率低的问题，提高了非机动车头盔佩戴检测的速度和准确度。