摘要::目的 为了解决变电所场景下因模型参数量过大、推理时间较长导致错检和漏检问题较高等问题,提出一种 基于知识蒸馏和注意力机制的变电所安全帽佩戴实时检测算法 YOLO-FE(BCKD)。 方法 为了减少模型的大小和 推理速度,使用 FasterNet block,去替换原本骨干网络和 NECK 部分中的 C2f 模块;为了减少因为降低模型参数而 导致精度大幅下降的问题,使用高效多尺度的 EMA 注意力机制加入骨干网络之中,提高模型提取特征的能力;最 后为了让模型能具有更高的精度,选择 YOLOv8m 作为教师网络,YOLO-FE 作为学生网络,使用 BCKD 知识蒸馏方 法对 YOLO-FE 进行知识蒸馏。 结果 该研究使用在煤矿变电所收集到的安全帽数据集为基础,选用不同算法进行 对比实验。 最终表明:改进的 YOLOv8 模型未蒸馏前平均精度达到 74. 5%,相较于 YOLOv8n 提高了 2. 5%,参数量 下降 26. 7%,检测速度提高 26%,经过蒸馏之后模型的平均精度达到 75. 4%。 为了验证算法的泛用性,在公开数据 集 GDUT-HWD 上测试,其精度值为 84. 2%,比原始模型提高 3%。 结论 文中提出的安全帽佩戴检测算法在精度和 实时性上取得了良好的平衡,满足变电所环境下的实时检测需求。