改进RT-DETR的油田人员异常行为检测OA
Improved RT-DETR for Abnormal Behavior Detection of Personnel in Oilfields
为解决油田作业现场因操作不规范导致的安全事故频发问题,提出了一种高效的面向油田复杂场景下异常行为检测的RT-DETR改进算法(HCH-DETR).首先,设计一种新型主干网络,结合双分支高频细节增强模块(HFERB)和CSP结构,提高模型高频细节特征提取能力,并有效减少模型计算量;其次,针对油田监控中目标尺度变化大、背景复杂等问题,提出一种基于上下文引导的空间特征重构特征金字塔网络(CGFRPN),通过矩形自校准注意力(RCA)增强多尺度特征融合,提高模型对多尺度目标的检测精度,并增强了其在复杂场景下的鲁棒性;最后,引入 Haar 小波下采样模块(H WD)优化传统下采样,提高模型对小目标的检测能力.在自建油田数据集上进行模型验证:mAP@0.5 和mAP@0.5:0.95 分别达到 85.4%和 55.1%,较原始RT-DETR模型提升 3.2 百分点和 2.4 百分点,同时计算量减少 7.1×109,参数量降低6.5×106;消融实验验证了各改进模块的有效性,泛化实验表明模型在VisDrone数据集上精度亦有提升.
To address the frequent safety accidents caused by non-standard operations at oilfield worksites,this study proposes an efficient improved RT-DETR algorithm,named HCH-DETR,for ab-normal behavior detection in complex oilfield scenarios.Firstly,a novel backbone network is designed by integrating the dual-branch High-Frequency Enhancement Residual Block(HFERB)and the Cross Stage Partial(CSP)structure.This design enhances the model's ability to extract high-frequency detailed features while effectively reducing the model's computational complexity.Secondly,to tackle the challen-ges such as large variations in target scale and complex backgrounds in oilfield monitoring,a Context-Guided Feature Reconstruction Feature Pyramid Network(CGFRPN)is proposed.It employs Rectangular Self-Calibration Attention(RCA)to strengthen multi-scale feature fusion,improving the detection accu-racy for multi-scale targets and bolstering robustness in complex scenes.Finally,the Haar Wavelet Down-sampling(HWD)module is introduced to optimize traditional downsampling,thereby improving the mod-el's capability for small-target detection.Experimental validation on a self-constructed oilfield dataset shows that the proposed model achieves 85.4%mAP@0.5 and 55.1%mAP@0.5:0.95,representing im-provements of 3.2 percentage points and 2.4 percentage points respectively over the original RT-DETR model.Meanwhile,the computational complexity is reduced by 7.1×109,and the number of parameters is decreased by 6.5×106.Ablation experiments verify the effectiveness of each improved module,and generalization experiments demonstrate that the model also achieves improved accuracy on the VisDrone dataset.
吴攀超;范文博;王婷婷
东北石油大学电气信息工程学院,黑龙江 大庆 163319东北石油大学电气信息工程学院,黑龙江 大庆 163319东北石油大学电气信息工程学院,黑龙江 大庆 163319
信息技术与安全科学
RT-DETR小目标检测特征提取异常行为深度学习
RT-DETRsmall target detectionfeature extractionabnormal behaviordeep learning
《机械与电子》 2026 (3)
32-40,46,10
国家自然科学基金资助项目(52474036)
评论