用于两轮车的头盔佩戴检测方法、装置、电子设备及介质

本申请涉及计算机视觉技术领域，公开了一种用于两轮车的头盔佩戴检测方法、装置、电子设备及介质，通过获取初始检测图像和头盔佩戴标签，根据头盔佩戴标签对初始检测图像进行标注，生成带有头盔佩戴标签的样本数据集，从样本数据集中提取样本训练集，通过预设的数据增强方式对样本训练集进行数据增强，得到目标数据集，通过在YOLOv5模型中引入K‑means聚类算法、加入SE注意力机制、引入EIOU损失函数，得到改进后的YOLOv5模型，基于目标数据集对改进后的YOLOv5模型进行训练，得到头盔佩戴检测模型，获取待测图像，通过头盔佩戴检测模型对待测图像进行头盔佩戴检测，得到头盔佩戴检测结果，以提供更实用、更准确的头盔佩戴检测方法。