机构地区:[1]北京石油化工学院信息工程学院,北京102617 [2]中国人民解放军93184部队,北京100076 [3]星禾环保科技有限公司,河南郑州450001
出 处:《图学学报》2025年第2期249-258,共10页Journal of Graphics
基 金:宁夏自然科学基金(2023AAC03889);北京市数字教育研究课题(BDEC2022619048);北京市教育委员会科研计划项目(KM202410017006)。
摘 要:柴油车辆排放黑烟是道路交通环保执法的重点和难点。由于受复杂环境条件的影响,针对目前黑烟检测存在精度和速度方面的不足,提出一种基于改进YOLOv8的轻量级柴油车辆排放黑烟的检测模型。首先,在YOLOv8主干网络的基础上,设计一种轻量化特征提取模块C2f-FasterRep提高模型的特征提取能力,同时C2f-FasterRep模块引入上下文锚框注意力机制模块来捕捉长距离的上下文信息,利用全局平均池化和条形卷积增强特征图中心区域的特征,从而提高检测精度;其次,在颈部部分提出一个新的网络结构用于融合主干网络提取的特征,并使用通道注意力模块和维度匹配机制对不同尺度的特征进行融合,增强了模型的多尺度特征融合能力;最后,使用Transformer解码器结构优化YOLOv8模型的检测头,同时,采用交并比感知的查询机制,有助于解码器查询的优化,提高了模型的分类和定位的性能。为保证实验的真实性和有效性,利用部署在河南许昌某道路断面的检测设备采集数据并进行测试验证。实验结果表明,该方法的mAp为95.4%,精确率为94.5%,召回率为97.5%,与现有的黑烟检测方法相比,具有更高的检测精度和更快的检测速度。消融实验结果表明该轻量化特征提取模块、特征融合模块和检测头有利于提高模型检测精度。The emission of black smoke from diesel vehicles is a critical and challenging issue in road traffic environmental protection enforcement.Due to the influence of complex environmental conditions,to address the limitations of existing black smoke detection methods in terms of accuracy and speed,a lightweight detection model of black smoke emission from diesel vehicles based on improved YOLOv8 was proposed.Firstly,based on the YOLOv8 backbone network,a lightweight feature extraction module C2f-FasterRep was designed to improve the feature extraction capability of the model,while the C2f-FasterRep module integrated a context-anchored attention mechanism module to capture long-range contextual information.The global average pooling and bar convolution were employed to enhance the features in the central region of the feature map,thereby improving the detection accuracy.Secondly,a new network structure was proposed in the neck part to fuse the features extracted from the backbone network.The channel attention module and dimensional matching mechanism were employed to fuse features of different scales,enhancing the model’s multi-scale feature fusion capability.Lastly,the detection head of the YOLOv8 model was optimized using a Transformer decoder structure.An intersection-parallel-ratio aware query mechanism was incorporated to optimize the decoder query,improving the model’s performance for classification and localization.In order to ensure the authenticity and effectiveness of the experiment,data were collected and tested and verified using the detection equipment deployed on a road section in Xuchang,Henan Province.The experimental results demonstrated that the mAp of the proposed method was achieved at 95.4%,the precision at 94.5%,and the recall rate at 97.5%.Compared with existing black smoke detection methods,the proposed approach exhibited higher detection accuracy and faster detection speed.The results of ablation experiments confirmed that the proposed lightweight feature extraction module,feature fusion m
关 键 词:目标检测 黑烟检测 YOLOv8 轻量级模型 注意力机制
分 类 号:TP391[自动化与计算机技术—计算机应用技术] X831[自动化与计算机技术—计算机科学与技术] X734[环境科学与工程—环境工程]
正在载入数据...
正在载入数据...
正在载入数据...
正在载入数据...
正在载入数据...
正在载入数据...
正在载入数据...