点击蓝字

关注我们

关注并星标

从此不迷路

计算机视觉研究院

公众号ID｜计算机视觉研究院

学习群｜扫码在主页获取加入方式

https://pmc.ncbi.nlm.nih.gov/articles/PMC13074888/pdf/sensors-26-02110.pdf

计算机视觉研究院专栏

Column of Computer Vision Institute

本文提出的轻量化 GSA-YOLO 算法，以极致轻量 + 超高精度，彻底攻克密闭空间安全帽检测痛点！

PART/1

    痛点   

电站密闭空间环境特殊，给视觉检测设下三道硬核关卡：

1. 光照极端：全黑、弱光、强光混杂，图像对比度低、细节模糊，模型特征提取直接失效；
2. 目标遮挡：管道、设备密集，工人头盔常被半遮挡，特征残缺易引发误检；
3. 小目标难测：远距离、俯拍视角，头盔占像素极少，常规模型频繁漏检；
4. 部署受限：现有模型参数庞大、算力要求高，无法在现场边缘设备实现实时运行。

PART/2

    创新   

破局创新：GSA-YOLO 四大改进，轻量与精度双在线

本研究基于 YOLOv8n 轻量化基线，打造GSA-YOLO（Ghost - 小目标 - 注意力） 算法，通过四大核心改进，兼顾轻量性与复杂场景鲁棒性。

【GSA-YOLO 整体架构图】

1. 骨干网升级：GCA-C2f 模块，抗光 + 降参双效替换原 C2f 模块为GCA-C2f，融合GhostConv 轻量化卷积（大幅减少参数）+CBAM 注意力机制（强化光照鲁棒性），既砍掉冗余计算，又精准捕捉复杂光照下的头盔特征。

2. 颈部优化：I-ECA 改进注意力，专治遮挡提出改进型 I-ECA 通道注意力，采用双分支特征扩展设计，不增加模型参数，却能强化遮挡目标的特征判别力，有效压制背景干扰。

3. 检测头增强：P2 分支，从根源解决小目标漏检新增P2 高分辨率检测分支（160×160），保留浅层细粒度空间信息，大幅提升小尺寸头盔的检测灵敏度，彻底告别小目标漏检问题。

4. 损失函数替换：WIoU，定位更稳、收敛更快用WIoU 损失替代原 CIoU 损失，动态调整难易样本权重，弱光、模糊目标的定位精度显著提升，模型训练收敛速度更快。

PART/3

     实验   

实验实证：数据碾压基线，性能全面领跑

研究自建 5000 张密闭空间安全帽数据集，覆盖光照剧变、小目标、弱光、遮挡四大场景，实验结果硬核亮眼：

1.核心指标爆表GSA-YOLO 仅2.3M 参数，mAP@0.5 达91.2%，较 YOLOv8n 基线精度提升 2.9%，参数减少 23.6%，推理速度达 92.6FPS！

【目标检测算法对比实验结果】

2.消融实验：每项改进都实打实有用四大模块协同发力，单模块改进均带来精度 / 效率提升，最终组合实现最优效果。

【消融实验结果】

3.可视化效果：复杂场景稳赢

• 光照剧变场景：置信度更高，无漏检误检；

  【光照变化场景检测对比】

  

• 小目标场景：彻底解决基线模型漏检问题；

  【小目标检测对比】

  

• 遮挡 / 多目标场景：弱光 + 遮挡叠加下仍精准识别。

  【遮挡多目标检测对比】

  

  【不同模型 mAP@0.5 曲线对比】

  

PART/4

     总结与展望   

应用展望：全场景密闭空间安全监测新方案

1. 落地优势：轻量易部署，边缘设备实时跑参数仅 2.3M，算力需求低，完美适配现场边缘终端，真正实现密闭空间安全帽实时监测。

2. 场景拓展：不止电站，全行业通用可快速迁移至矿山、地下管廊、隧道等各类密闭空间作业场景，通用性拉满。

3. 未来方向融合夜视相机强化弱光采集能力，拓展至防护服、安全带等全品类劳保用品检测，打造一站式工业安全监测系统。

工业安全无小事，轻量化视觉检测技术为密闭空间作业筑牢安全防线。GSA-YOLO 以小参数、大精度、强适配的特性，成为工业安全帽实时检测的优选方案！

有相关需求的你可以联系我们！

END

转载请联系本公众号获得授权

计算机视觉研究院学习群等你加入！

ABOUT

计算机视觉研究院

计算机视觉研究院主要涉及深度学习领域，主要致力于目标检测、目标跟踪、图像分割、OCR、模型量化、模型部署等研究方向。研究院每日分享最新的论文算法新框架，提供论文一键下载，并分享实战项目。研究院主要着重”技术研究“和“实践落地”。研究院会针对不同领域分享实践过程，让大家真正体会摆脱理论的真实场景，培养爱动手编程爱动脑思考的习惯！

往期推荐 

🔗

• YOLO-TLA：一种基于 YOLOv5 的高效轻量级小目标检测模型

• ViT-YOLO：基于Transformer的用于目标检测的YOLO算法

• SSMA-YOLO：一种轻量级的 YOLO 模型，具备增强的特征提取与融合能力，适用于无人机航拍的船舶图像检测

• LUD-YOLO：一种用于无人机的新型轻量级目标检测网络

• Gold-YOLO：基于聚合与分配机制的高效目标检测器

• Drone-YOLO：一种有效的无人机图像目标检测

• 「无人机+AI」“空中城管”

• 无人机+AI：光伏巡检自动化解决方案

• 无人机视角下多类别船舶检测及数量统计
  

• 机场项目：解决飞行物空间大小/纵横比、速度、遮挡等问题引起的实时目标检测问题

• 2PCNet：昼夜无监督域自适应目标检测（附原代码）

• YOLO-S：小目标检测的轻量级、精确的类YOLO网络

• 大改Yolo框架 |  能源消耗极低的目标检测新框架（附论文下载）

• 改进的检测算法：用于高分辨率光学遥感图像目标检测