###
计算机系统应用英文版:2023,32(2):217-225
本文二维码信息
码上扫一扫!
多尺度特征融合的雾霾环境下车辆检测
(1.湖南工业大学 轨道交通学院, 株洲 412007;2.中国电子科技集团公司 第十五研究所, 北京 100089)
Multi-scale Feature Fusion for Vehicle Detection in Haze Environment
(1.College of Railway Transportation, Hunan University of Technology, Zhuzhou 412007, China;2.The 15th Research Institute, China Electronics Technology Group Corporation, Beijing 100089, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 483次   下载 878
Received:June 30, 2022    Revised:August 09, 2022
中文摘要: 针对雾霾环境下车辆检测准确率低、漏检严重的问题, 提出一种多尺度特征融合的雾霾环境下车辆检测算法. 首先利用条件生成对抗网络对雾霾图像进行去雾预处理, 然后针对雾霾环境下目标特征不明显的特点, 提出多尺度特征融合模块, 在YOLOv3的基础上, 从主干网络提取特征时增加一条浅层分支和深层特征进行上采样拼接融合, 得到尺度为104×104的特征图, 用于增强浅层的语义信息. 并采用CBAM注意力机制引导下的特征增强策略, 保证上下文信息的完整性, 以提高检测的精度, 最后将去雾后图片送入改进后的YOLOv3网络进行检测. 实验结果表明, 相较于原始网络, 该算法在RTTS数据集上的检测结果更加优秀, 模型可以达到81%的平均精度和67.52%的召回率, 能够更加精确的定位到车辆.
Abstract:Given low vehicle detection accuracy and serious miss detection in a haze environment, a vehicle detection algorithm with multi-scale feature fusion in a haze environment is proposed. Firstly, the conditional generation and adversarial network is employed to preprocess the haze images. Then, as the object feature is not obvious in a haze environment, a multi-scale feature fusion module is put forward. On the basis of YOLOv3, a shallow branch is added for upsampling splicing and fusing with deep layer features during extracting features from backbone networks. As a result, the feature map with the scale of 104×104 is obtained, which is adopted to enhance the shallow semantic information. The feature enhancement strategy guided by the CBAM attention mechanism is utilized to ensure the integrity of context information and improve detection accuracy. Finally, the dehazed images are sent to the improved YOLOv3 network for detection. Experimental results show that the proposed algorithm has better performance than the YOLOv3 algorithm on the RTTS dataset. The proposed model can achieve an average accuracy of 81% and a recall of 67.52% and can locate vehicles more accurately.
文章编号:     中图分类号:    文献标志码:
基金项目:国家重点研发计划(2021YFF0501102); 湖南省自然科学基金(2020JJ5128); 湖南省高新技术产业科技创新引领计划(2021GK4010)
引用文本:
王忠美,薛子豪,伍宣衡,郑良.多尺度特征融合的雾霾环境下车辆检测.计算机系统应用,2023,32(2):217-225
WANG Zhong-Mei,XUE Zi-Hao,WU Xuan-Heng,ZHENG Liang.Multi-scale Feature Fusion for Vehicle Detection in Haze Environment.COMPUTER SYSTEMS APPLICATIONS,2023,32(2):217-225