安全帽佩戴检测方法、模型训练方法、装置及电子设备

本申请涉及一种安全帽佩戴检测方法、训练方法、装置、设备及存储介质，其中，安全帽佩戴检测方法，包括：获取包含井下人员的待检测图像以及安全帽佩戴检测模型，所述安全帽佩戴检测模型为改进的YOLOv8模型，所述改进的YOLOv8模型为BackBone层中增加有基于全局上下文的通道注意力机制单元、空间重建单元以及通道重建单元的YOLOv8模型训练得到；将待检测图像输入改进的YOLOv8模型，得到安全帽佩戴检测结果；其中，所述基于全局上下文的通道注意力机制单元，用于对原特征图进行全局平均池化以获取全局信息；接着通过全连接层和ReLU激活函数学习各特征通道的重要性权重；然后通过图卷积网络来处理学习后的特征，并通过Sigmoid函数对权重进行映射处理得到特征图；将映射处理后的特征图与所述原特征图进行融合。通过本申请解决相关技术中存在的模型轻量化的同时难以保证检测精度的问题。