孙卉梅, 刘路, 王德刚, 王太勇. 面向高后果区工程车辆视觉检测的YOLO-MMCE算法[J]. 油气储运, 2024, 43(9): 1031-1038. DOI: 10.6047/j.issn.1000-8241.2024.09.008
引用本文: 孙卉梅, 刘路, 王德刚, 王太勇. 面向高后果区工程车辆视觉检测的YOLO-MMCE算法[J]. 油气储运, 2024, 43(9): 1031-1038. DOI: 10.6047/j.issn.1000-8241.2024.09.008
SUN Huimei, LIU Lu, WANG Degang, WANG Taiyong. YOLO-MMCE algorithm for visual detection of engineering vehicles in high-consequence area[J]. Oil & Gas Storage and Transportation, 2024, 43(9): 1031-1038. DOI: 10.6047/j.issn.1000-8241.2024.09.008
Citation: SUN Huimei, LIU Lu, WANG Degang, WANG Taiyong. YOLO-MMCE algorithm for visual detection of engineering vehicles in high-consequence area[J]. Oil & Gas Storage and Transportation, 2024, 43(9): 1031-1038. DOI: 10.6047/j.issn.1000-8241.2024.09.008

面向高后果区工程车辆视觉检测的YOLO-MMCE算法

YOLO-MMCE algorithm for visual detection of engineering vehicles in high-consequence area

  • 摘要:
    目的 高后果区大型施工现场的工程车辆给埋地管道带来严重的安全隐患。针对当前常用方法在工程车辆重叠目标检测和日照变化场景下的目标检测方面存在漏检率高、检测精度低的问题,以挖掘机、装载机、压路机及重型货车4类常见工程车辆作为识别对象,提出了一种基于改进YOLOv5的工程车辆目标检测方法——YOLO-MMCE。
    方法 采用Mosaic+Mixup结合的数据增强方式,增强对不同场景的适应能力,提高模型在实际复杂环境和模糊情况下的鲁棒性和泛化性。针对目标重叠和光照变化导致的特征不明显问题,在YOLOv5网络模型中引入坐标注意力(Coordinate Attention, CA)机制,增强网络模型的特征提取能力;为了提升预测边框回归精度,引入了高效率交并比(Efficient Intersection over Union, EIOU)函数,计算预测框与真实框的宽高差异值并取代纵横比,进一步提高算法检测精度。
    结果 以兰郑长成品油管道高后果区监控摄像机获取的施工现场照片为数据集,对YOLO-MMCE算法进行验证。结果表明,对YOLOv5算法3个方面的改进均能提高其在实际工况下工程车辆目标检测的精度,总体平均精度均值(Mean Average Precision, mAP)达到84.8%,比原始YOLOv5算法提高了6.9%。对挖掘机、装载机、压路机及重型货车的目标检测mAP分别提高了4.4%、7.5%、9.5%、6.0%。
    结论 YOLO-MMCE算法有效解决了重叠目标检测和日照变化场景下的工程车辆目标检测问题,具备实际应用价值。

     

    Abstract:
    Objective Engineering vehicles operating on large-scale construction sites in high-consequence areas pose severe safety hazards to buried pipelines. This paper addresses the shortcomings of current common techniques for detecting overlapped targets of engineering vehicles and target detection in scenarios with varying sunlight, highlighting issues such as high miss rates and low detection accuracies. The paper introduces a target detection method for engineering vehicles named YOLO-MMCE, which is based on an improved version of YOLOv5. This method focuses on recognizing four main types of engineering vehicles: excavators, loaders, rollers, and heavy trucks.
    Methods The Mosaic + Mixup combined data augmentation approach was adopted to improve the model's adaptability to diverse scenarios and strengthen its robustness and generalization in intricate real-world settings and ambiguous conditions. In response to challenges pertaining to overlapping targets and inconspicuous features due to illumination variations, a coordinate attention (CA) mechanism was integrated into the YOLOv5 network model to amplify its feature extraction capacity. Additionally, to improve the regression accuracy of prediction borders, an Efficient Intersection over Union (EIOU) function was incorporated to calculate the width-height difference between prediction and real borders to replace the aspect ratio, thus further elevating the detection accuracy of the algorithm.
    Results The YOLO-MMCE algorithm was validated using the datasets comprised of construction site photos captured by surveillance cameras in the high-consequence areas along the Lanzhou-Zhengzhou-Changsha product oil pipeline. The results revealed precision enhancements in engineering vehicle target detection under real-world conditions, resulting from the YOLOv5 algorithm improvements in three aspects. These advancements led to an overall Mean Average Precision (mAP) of 84.8%, a 6.9% increase over the original YOLOv5 algorithm. The target detection mAPs for excavators, loaders, rollers, and heavy trucks were raised by 4.4%, 7.5%, 9.5%, and 6.0% respectively.
    Conclusion The YOLO-MMCE algorithm provides an efficient solution for detecting overlapped targets and engineering vehicle targets in environments with varying sunlight conditions, illustrating its values in practical applications.

     

/

返回文章
返回