基于跨尺度特征融合的泵站安全帽检测方法.pdfVIP

测控技术

·１６·２０２３４２７模式识别与人工智能

年第卷第期

基于跨尺度特征融合的泵站安全帽检测方法

１１２，３，４２，３，４

，，，

李记恒褚霄杨王涛刘鹏宇

（１．北京市南水北调团城湖管理处，北京１００１９５；２．北京工业大学信息学部，北京１００１２４；

３．１００１２４４．１００１２４

先进信息网络北京实验室，北京；北京工业大学计算智能与智能系统北京市重点实验室，北京）

摘要为应对泵站场景下设备和人员之间目标被遮挡及远距离小目标对泵站重点区域安全帽佩戴自动

：

监管带来的挑战提出了一种融合注意力机制和跨尺度特征融合的安全帽佩戴检测算法以克服在远距

，，

离有遮挡场景下安全帽检测准确度低的问题通过采集泵站监控视频数据构建泵站场景安全帽数据

、。

集在特征提取网络中加入注意力机制模块使得模型更关注于小目标的通道信息同时增加检测层使

，，；

，（，）

得特征融合时能结合多级特征并使用柔和非极大值抑制ＳｏｆｔＮｏｎＭａｎｉｍｕｍＳｕｐｐｒｅｓｓｉｏｎＳｏｆｔＮＭＳ和

完全交并比算法进行改进以减少遮挡目标漏检情况在自建数

（ＣｏｍｐｌｅｔｅＩｎｔｅｒｓｅｃｔｉｏｎｏｖｅｒＵｎｉｏｎ，ＣＩｏＵ）。

据集进行试验，结果表明改进后的算法平均准确率达到９３５％，与其他目标检测算法相比精度均有所

提升证明该方法在泵站重点区域场景安全帽检测任务中具有良好的性能

，。

关键词：泵站；安全帽检测；注意力机制；特征融合

中图分类号ＴＰ２９文献标志码Ａ文章编号１０００－８８２９２０２３０７－００１６－０６

：：：（）

ｄｏｉ：１０．１９７０８／ｊ．ｃｋｊｓ．２０２３．０７．００３

ＳａｆｅｔｙＨｅｌｍｅｔＤｅｔｅｃｔｉｏｎＭｅｔｈｏｄｏｆＰｕｍｐＳｔａｔｉｏｎＢａｓｅｄｏｎＣｒｏｓｓＳｃａｌｅ

ＦｅａｔｕｒｅＦｕｓｉｏｎ

１１２牞３牞４２牞３牞４

ＬＩＪｉｈｅｎｇ牞ＣＨＵＸｉａｏｙａｎｇ牞ＷＡＮＧＴａｏ牞ＬＩＵＰｅｎｇｙｕ

牗１．ＴｕａｎｃｈｅｎｇＬａｋｅＭａｎａｇｅｍｅｎｔＯｆｆｉｃｅｏｆＢｅｉｊｉｎｇＳｏｕｔｈｔｏＮｏｒｔｈＷａｔｅｒＴｒａｎｓｆｅｒＰｒｏｊｅｃｔ牞Ｂｅｉｊ