[1]院老虎,常玉坤,刘家夫.基于改进YOLOv5s的雾天场景车辆检测方法[J].郑州大学学报(工学版),2023,44(03):37-43.[doi:10.13705/j.issn.1671-6833.2023.03.005]
　YUAN Laohu,CHANG Yukun,LIU Jiafu.Vehicle Detection Method Based on Improved YOLOv5s in Foggy Scene[J].Journal of Zhengzhou University (Engineering Science),2023,44(03):37-43.[doi:10.13705/j.issn.1671-6833.2023.03.005]
点击复制

基于改进YOLOv5s的雾天场景车辆检测方法()
分享到：

《郑州大学学报(工学版)》[ISSN:1671-6833/CN:41-1339/T]

卷:
44卷
期数:
2023年03期
页码:
37-43
栏目:
出版日期:
2023-04-30

文章信息/Info

Title:
Vehicle Detection Method Based on Improved YOLOv5s in Foggy Scene
作者:
院老虎; 常玉坤; 刘家夫
沈阳航空航天大学 航空宇航学院,辽宁 沈阳 110136

Author(s):
YUAN Laohu; CHANG Yukun; LIU Jiafu
Shenyang University of Aeronautics and Astronautics University of Aeronautics and Astronautics, Liaoning Shenyang 110136

关键词:
深度学习;  YOLOv5s;  车辆检测;  数据增强;  雾气模拟
Keywords:
deep learning;  YOLOv5s;  vehicle detection;  data augmentation;  fog simulation
DOI:
10.13705/j.issn.1671-6833.2023.03.005
文献标志码:
A
摘要:
为了解决现有的目标检测方法在雾天场景下存在识别准确率低、易漏检的问题，提出一种改进 YOLOv5s 的雾天车辆检测方法。首先，以 VisDrone 数据集为基础，通过大气散射模型生成轻雾数据集( LightFogVisDrone) 和浓雾数据集( ThickFogVisDrone) ，并收集真实雾天场景图片组成混合浓度数据集( MixFogData) ; 其 次，对 原 始 YOLOv5s 的 Mosaic 数据增强方式进行改进，由原始 的 4 张 图 片 改 为 9 张图片进行随机剪切，减 少 灰 色 背 景面积，加快模型收敛，提高训练效率，在预测端之前添加 CBAM 注 意 力 机 制，以此来增强模型的图像特 征提取能力，改善遮挡目标与小目标的漏检问题; 最 后，优 化 NMS 非极大抑制值先验框，改善车辆目标的 漏检问题。实验结果表明: 与原 始 YOLOv5s 相 比，改 进 YOLOv5s 在 轻 雾、浓雾和混合雾气状态下的平均 精确率分别提高了 16. 14、16. 16 和 15. 05 百分点。改进 YOLOv5s 对于雾天环境下车辆目标的检测具有 有效性和实用性。
Abstract:
In order to solve the problems of low recognition accuracy and easy omission of existing target detection methods in foggy scenes，an improved vehicle detection method based on YOLOv5s was proposed． Firstly，based on VisDrone data set，LightFogVisDrone and ThickFogVisDrone were generated by atmospheric scattering model，and the MixFogData was composed of real fog scene pictures． Secondly，the Mosaic data enhancement method of the original model was improved from the original 4 pictures to 9 pictures randomly，which reduced the gray background area，accelerated the convergence of the model and improved the training efficiency，and the CBAM attention mechanism module was added before the prediction end to improve the feature extraction ability of the network to tackle the problem of missed detection of occluded targets and small targets． Finally，the prior frame of NMS nonmaximum suppression value was optimized to improve the problem of missing detection of vehicle targets． The experimental results showed that，compared with the original YOLOv5s，the average accuracy of the improved YOLOv5s in light fog，dense fog and mixed fog was increased by 16. 14，16. 16 and 15. 05 percentage points，respectively，which proved that the improved YOLOv5s was effective and practical for vehicle target detection in foggy environment．

相似文献/References:

[1]袁航,钟发海,聂上上,等.基于卷积神经网络的道路拥堵识别研究[J].郑州大学学报(工学版),2019,40(02):21.[doi:10.13705/j.issn.1671-6833.2019.02.008]
　LUO Ronghui,YUAN Hang,ZHONG Fahai,et al.The Research of Traffic Jam Detection Based on Convolutional Neural Network[J].Journal of Zhengzhou University (Engineering Science),2019,40(03):21.[doi:10.13705/j.issn.1671-6833.2019.02.008]
[2]朱俊丞,杨之乐,郭媛君,等.深度学习在电力负荷预测中的应用综述[J].郑州大学学报(工学版),2019,40(05):12.[doi:10.13705/j.issn.1671-6833.2019.05.005]
　Zhu Juncheng,Young Joy,Guo Yuanjun,et al.A review of the application of deep learning in power load forecasting[J].Journal of Zhengzhou University (Engineering Science),2019,40(03):12.[doi:10.13705/j.issn.1671-6833.2019.05.005]
[3]黄文锋,徐珊珊,孙燚,等.基于多分辨率卷积神经网络的火焰检测[J].郑州大学学报(工学版),2019,40(05):79.[doi:10.13705/j.issn.1671-6833.2019.05.022]
　Huang Wenfeng,Susan Hsu,Sun Yi,et al.Fire Detection Based on Multi-resolution Convolution Neural Network in Various Scenes[J].Journal of Zhengzhou University (Engineering Science),2019,40(03):79.[doi:10.13705/j.issn.1671-6833.2019.05.022]
[4]陈义飞、郭胜、潘文安、陆彦辉.基于多源传感器数据融合的三维场景重建[J].郑州大学学报(工学版),2021,42(02):81.[doi:10.13705/j.issn.1671-6833.2021.02.008]
　Chen Yifei,Guo Sheng,Pan Wenan,et al.3D Scene Reconstruction ba＜x＞sed On Multi-source Sensor Data Fusion[J].Journal of Zhengzhou University (Engineering Science),2021,42(03):81.[doi:10.13705/j.issn.1671-6833.2021.02.008]
[5]李学相,曹淇,刘成明.基于无配对生成对抗网络的图像超分辨率重建[J].郑州大学学报(工学版),2021,42(05):1.[doi:10.13705/j.issn.1671-6833.2021.05.018]
　Li Xuexiang,Cao Qi,Liu Chengming,et al.Image super-resolution ba＜x＞sed on no match generative adversarial network[J].Journal of Zhengzhou University (Engineering Science),2021,42(03):1.[doi:10.13705/j.issn.1671-6833.2021.05.018]
[6]王希鹏,李永,李智,等.融合图像深度的抗遮挡目标跟踪算法[J].郑州大学学报(工学版),2021,42(05):19.[doi:10.13705/j.issn.1671-6833.2021.05.011]
　Wang Xipeng,Li Yong,Li Zhi,et al.Anti-occlusion target tracking algorithm ba＜x＞sed on image depth—ncig2020[J].Journal of Zhengzhou University (Engineering Science),2021,42(03):19.[doi:10.13705/j.issn.1671-6833.2021.05.011]
[7]卢晨辉,冯硕,易爱华,等.基于深度学习的加油站销量预测与营销策略应用研究[J].郑州大学学报(工学版),2022,43(01):1.[doi:10.13705/j.issn.1671-6833.2022.01.014]
　Lu Chenhui,Feng Shuo,Yi Aihua,et al.A Deep Learning ba＜x＞sed Gasoline Station Sales Prediction and its Application on Promotion Strategy[J].Journal of Zhengzhou University (Engineering Science),2022,43(03):1.[doi:10.13705/j.issn.1671-6833.2022.01.014]
[8]陈浩杰,黄锦,左兴权,等.基于宽度&深度学习的基站网络流量预测方法[J].郑州大学学报(工学版),2022,43(01):7.[doi:10.13705/j.issn.1671-6833.2022.01.011]
　Chen Haojie,Huang Jin,Zuo Xingquan,et al.ba＜x＞se Station Network Traffic Prediction Method ba＜x＞sed on Wide and Deep Learning[J].Journal of Zhengzhou University (Engineering Science),2022,43(03):7.[doi:10.13705/j.issn.1671-6833.2022.01.011]
[9]成科扬,荣兰,蒋森林,等.基于深度学习的遥感图像超分辨率重建技术综述[J].郑州大学学报(工学版),2022,43(05):8.[doi:10.13705/j.issn.1671-6833.2022.05.013]
　CHENG K Y,RONG L,JIANG S L,et al.CCFAI: Survey Of Deep-Learning Approaches For Remote Sensing Super-Resolution Reconstruction[J].Journal of Zhengzhou University (Engineering Science),2022,43(03):8.[doi:10.13705/j.issn.1671-6833.2022.05.013]

更新日期/Last Update: 2023-05-08