改进YOLOv7算法及其油田生产违规行为检测OA
Improved YOLOv7 Algorithm for Oilfield Regulatory Violation Detection
针对油田现场监控中摄像头安装高度较高、目标体积较小导致检测难度大的问题,该文提出了一种改进的YOLOv7 目标检测算法.首先,在网络结构中引入GD融合机制,通过高效的信息聚合策略,动态整合来自不同层级的特征图信息,从而增强模型对多尺度目标,尤其是小目标的检测能力.其次,加入Biformer模块,利用其双分支路由注意力机制从全局视角分析特征间的相关性,有效过滤背景干扰和冗余特征,提升模型对关键目标区域的关注度,同时结合注意力稀疏化策略降低冗余计算,兼顾检测精度与计算效率.最后,将传统的 IoU 损失替换为 ICoU-NWD 损失函数,引入Wasserstein距离作为边界框之间的度量方式,使边界框预测更准确,尤其在面对尺度变化较大的目标时更具鲁棒性.实验结果表明,改进模型在典型油田场景下的mAP达到97.0%,比原始YOLOv7 提升了7.2%;在提升检测准确率和特征表达能力的同时,参数量仅增加12.1%,计算量仅上升3.7%,适合部署在边缘计算设备上,满足复杂环境下的智能检测需求.
To address the challenges in oilfield site monitoring caused by high camera installation angles and small target sizes,we propose an improved YOLOv7 object detection algorithm.Firstly,a GD fusion mechanism is introduced into the neck network to integrate feature maps from different levels,enhancing detection capability for multi-scale targets,especially small ones.Secondly,the Biformer module is added,and its dual-branch routing attention mechanism is utilized to analyze the correlation between features from a global perspective,effectively filtering out background interference and redundant features,enhancing the model's attention to key target areas.At the same time,the attention sparsity strategy is combined to reduce redundant computations,balancing detection accuracy and computational efficiency.Finally,the standard IoU loss is replaced with ICoU-NWD loss,which uses Wasserstein distance for more accurate bounding box localization,especially for targets with large scale variations.Experimental results show that the mAP of the improved model in typical oilfield scenarios reaches 97.0%,which is 7.2%higher than that of the original YOLOv7.While enhancing the detection accuracy and feature expression capabilities,the number of parameters only increases by 12.1%,and the computational load only rises by 3.7%.It is suitable for deployment on edge computing devices to meet the intelligent detection requirements in complex environments.
任伟建;李虞龙;康朝海;霍凤财;任璐;张永丰
东北石油大学 电气信息工程学院,黑龙江 大庆 163318||东北石油大学 黑龙江省网络化与智能控制重点实验室,黑龙江 大庆 163318东北石油大学 电气信息工程学院,黑龙江 大庆 163318东北石油大学 电气信息工程学院,黑龙江 大庆 163318||东北石油大学 黑龙江省网络化与智能控制重点实验室,黑龙江 大庆 163318东北石油大学 电气信息工程学院,黑龙江 大庆 163318||东北石油大学 黑龙江省网络化与智能控制重点实验室,黑龙江 大庆 163318海洋石油工程股份有限公司,天津 300450大庆油田有限责任公司 第二采油厂规划设计研究所,黑龙江 大庆 163318
信息技术与安全科学
改进YOLOv7算法小目标检测GD融合机制BiformerWasserstein距离ICoU-NWD
improved YOLOv7 algorithmsmall object detectionGD fusion mechanismBiformerWasserstein distanceICoU-NWD
《计算机技术与发展》 2026 (1)
156-161,6
河北省自然科学基金面上项目(D2022107001)
评论