Cargando…

Enhanced YOLOv5: An Efficient Road Object Detection Method

Accurate identification of road objects is crucial for achieving intelligent traffic systems. However, developing efficient and accurate road object detection methods in complex traffic scenarios has always been a challenging task. The objective of this study was to improve the target detection algo...

Descripción completa

Detalles Bibliográficos
Autores principales: Chen, Hao, Chen, Zhan, Yu, Hang
Formato: Online Artículo Texto
Lenguaje:English
Publicado: MDPI 2023
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10611198/
https://www.ncbi.nlm.nih.gov/pubmed/37896450
http://dx.doi.org/10.3390/s23208355
_version_ 1785128435109068800
author Chen, Hao
Chen, Zhan
Yu, Hang
author_facet Chen, Hao
Chen, Zhan
Yu, Hang
author_sort Chen, Hao
collection PubMed
description Accurate identification of road objects is crucial for achieving intelligent traffic systems. However, developing efficient and accurate road object detection methods in complex traffic scenarios has always been a challenging task. The objective of this study was to improve the target detection algorithm for road object detection by enhancing the algorithm’s capability to fuse features of different scales and levels, thereby improving the accurate identification of objects in complex road scenes. We propose an improved method called the Enhanced YOLOv5 algorithm for road object detection. By introducing the Bidirectional Feature Pyramid Network (BiFPN) into the YOLOv5 algorithm, we address the challenges of multi-scale and multi-level feature fusion and enhance the detection capability for objects of different sizes. Additionally, we integrate the Convolutional Block Attention Module (CBAM) into the existing YOLOv5 model to enhance its feature representation capability. Furthermore, we employ a new non-maximum suppression technique called Distance Intersection Over Union (DIOU) to effectively address issues such as misjudgment and duplicate detection when significant overlap occurs between bounding boxes. We use mean Average Precision (mAP) and Precision (P) as evaluation metrics. Finally, experimental results on the BDD100K dataset demonstrate that the improved YOLOv5 algorithm achieves a 1.6% increase in object detection mAP, while the P value increases by 5.3%, effectively improving the accuracy and robustness of road object recognition.
format Online
Article
Text
id pubmed-10611198
institution National Center for Biotechnology Information
language English
publishDate 2023
publisher MDPI
record_format MEDLINE/PubMed
spelling pubmed-106111982023-10-28 Enhanced YOLOv5: An Efficient Road Object Detection Method Chen, Hao Chen, Zhan Yu, Hang Sensors (Basel) Article Accurate identification of road objects is crucial for achieving intelligent traffic systems. However, developing efficient and accurate road object detection methods in complex traffic scenarios has always been a challenging task. The objective of this study was to improve the target detection algorithm for road object detection by enhancing the algorithm’s capability to fuse features of different scales and levels, thereby improving the accurate identification of objects in complex road scenes. We propose an improved method called the Enhanced YOLOv5 algorithm for road object detection. By introducing the Bidirectional Feature Pyramid Network (BiFPN) into the YOLOv5 algorithm, we address the challenges of multi-scale and multi-level feature fusion and enhance the detection capability for objects of different sizes. Additionally, we integrate the Convolutional Block Attention Module (CBAM) into the existing YOLOv5 model to enhance its feature representation capability. Furthermore, we employ a new non-maximum suppression technique called Distance Intersection Over Union (DIOU) to effectively address issues such as misjudgment and duplicate detection when significant overlap occurs between bounding boxes. We use mean Average Precision (mAP) and Precision (P) as evaluation metrics. Finally, experimental results on the BDD100K dataset demonstrate that the improved YOLOv5 algorithm achieves a 1.6% increase in object detection mAP, while the P value increases by 5.3%, effectively improving the accuracy and robustness of road object recognition. MDPI 2023-10-10 /pmc/articles/PMC10611198/ /pubmed/37896450 http://dx.doi.org/10.3390/s23208355 Text en © 2023 by the authors. https://creativecommons.org/licenses/by/4.0/Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
spellingShingle Article
Chen, Hao
Chen, Zhan
Yu, Hang
Enhanced YOLOv5: An Efficient Road Object Detection Method
title Enhanced YOLOv5: An Efficient Road Object Detection Method
title_full Enhanced YOLOv5: An Efficient Road Object Detection Method
title_fullStr Enhanced YOLOv5: An Efficient Road Object Detection Method
title_full_unstemmed Enhanced YOLOv5: An Efficient Road Object Detection Method
title_short Enhanced YOLOv5: An Efficient Road Object Detection Method
title_sort enhanced yolov5: an efficient road object detection method
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10611198/
https://www.ncbi.nlm.nih.gov/pubmed/37896450
http://dx.doi.org/10.3390/s23208355
work_keys_str_mv AT chenhao enhancedyolov5anefficientroadobjectdetectionmethod
AT chenzhan enhancedyolov5anefficientroadobjectdetectionmethod
AT yuhang enhancedyolov5anefficientroadobjectdetectionmethod