Cargando…

Pedestrian detection algorithm integrating large kernel attention and YOLOV5 lightweight model

In the context of intelligent driving, pedestrian detection faces challenges related to low accuracy in target recognition and positioning. To address this issue, a pedestrian detection algorithm is proposed that integrates a large kernel attention mechanism with the YOLOV5 lightweight model. The al...

Descripción completa

Detalles Bibliográficos
Autores principales: Yin, Yuping, Zhang, Zheyu, Wei, Lin, Geng, Chao, Ran, Haoxiang, Zhu, Haodong
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Public Library of Science 2023
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10686420/
https://www.ncbi.nlm.nih.gov/pubmed/38019827
http://dx.doi.org/10.1371/journal.pone.0294865
_version_ 1785151772396879872
author Yin, Yuping
Zhang, Zheyu
Wei, Lin
Geng, Chao
Ran, Haoxiang
Zhu, Haodong
author_facet Yin, Yuping
Zhang, Zheyu
Wei, Lin
Geng, Chao
Ran, Haoxiang
Zhu, Haodong
author_sort Yin, Yuping
collection PubMed
description In the context of intelligent driving, pedestrian detection faces challenges related to low accuracy in target recognition and positioning. To address this issue, a pedestrian detection algorithm is proposed that integrates a large kernel attention mechanism with the YOLOV5 lightweight model. The algorithm aims to enhance long-term attention and dependence during image processing by fusing the large kernel attention module with the C3 module. Furthermore, it addresses the lack of long-distance relationship information in channel and spatial feature extraction and representation by introducing the Coordinate Attention mechanism. This mechanism effectively extracts local information and focused location details, thereby improving detection accuracy. To improve the positioning accuracy of obscured targets, the alpha CIOU bounding box regression loss function is employed. It helps mitigate the impact of occlusions and enhances the algorithm’s ability to precisely localize pedestrians. To evaluate the effectiveness of trained model, experiments are conducted on the BDD100K pedestrian dataset as well as the Pascal VOC dataset. Experimental results demonstrate that the improved attention fusion YOLOV5 lightweight model achieves an average accuracy of 60.3%. Specifically, the detection accuracy improves by 1.1% compared to the original YOLOV5 algorithm, and the accuracy performance index reaches 73.0%. These findings strongly indicate the proposed algorithm in significantly enhancing the accuracy of pedestrian detection in road scenes.
format Online
Article
Text
id pubmed-10686420
institution National Center for Biotechnology Information
language English
publishDate 2023
publisher Public Library of Science
record_format MEDLINE/PubMed
spelling pubmed-106864202023-11-30 Pedestrian detection algorithm integrating large kernel attention and YOLOV5 lightweight model Yin, Yuping Zhang, Zheyu Wei, Lin Geng, Chao Ran, Haoxiang Zhu, Haodong PLoS One Research Article In the context of intelligent driving, pedestrian detection faces challenges related to low accuracy in target recognition and positioning. To address this issue, a pedestrian detection algorithm is proposed that integrates a large kernel attention mechanism with the YOLOV5 lightweight model. The algorithm aims to enhance long-term attention and dependence during image processing by fusing the large kernel attention module with the C3 module. Furthermore, it addresses the lack of long-distance relationship information in channel and spatial feature extraction and representation by introducing the Coordinate Attention mechanism. This mechanism effectively extracts local information and focused location details, thereby improving detection accuracy. To improve the positioning accuracy of obscured targets, the alpha CIOU bounding box regression loss function is employed. It helps mitigate the impact of occlusions and enhances the algorithm’s ability to precisely localize pedestrians. To evaluate the effectiveness of trained model, experiments are conducted on the BDD100K pedestrian dataset as well as the Pascal VOC dataset. Experimental results demonstrate that the improved attention fusion YOLOV5 lightweight model achieves an average accuracy of 60.3%. Specifically, the detection accuracy improves by 1.1% compared to the original YOLOV5 algorithm, and the accuracy performance index reaches 73.0%. These findings strongly indicate the proposed algorithm in significantly enhancing the accuracy of pedestrian detection in road scenes. Public Library of Science 2023-11-29 /pmc/articles/PMC10686420/ /pubmed/38019827 http://dx.doi.org/10.1371/journal.pone.0294865 Text en © 2023 Yin et al https://creativecommons.org/licenses/by/4.0/This is an open access article distributed under the terms of the Creative Commons Attribution License (https://creativecommons.org/licenses/by/4.0/) , which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.
spellingShingle Research Article
Yin, Yuping
Zhang, Zheyu
Wei, Lin
Geng, Chao
Ran, Haoxiang
Zhu, Haodong
Pedestrian detection algorithm integrating large kernel attention and YOLOV5 lightweight model
title Pedestrian detection algorithm integrating large kernel attention and YOLOV5 lightweight model
title_full Pedestrian detection algorithm integrating large kernel attention and YOLOV5 lightweight model
title_fullStr Pedestrian detection algorithm integrating large kernel attention and YOLOV5 lightweight model
title_full_unstemmed Pedestrian detection algorithm integrating large kernel attention and YOLOV5 lightweight model
title_short Pedestrian detection algorithm integrating large kernel attention and YOLOV5 lightweight model
title_sort pedestrian detection algorithm integrating large kernel attention and yolov5 lightweight model
topic Research Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10686420/
https://www.ncbi.nlm.nih.gov/pubmed/38019827
http://dx.doi.org/10.1371/journal.pone.0294865
work_keys_str_mv AT yinyuping pedestriandetectionalgorithmintegratinglargekernelattentionandyolov5lightweightmodel
AT zhangzheyu pedestriandetectionalgorithmintegratinglargekernelattentionandyolov5lightweightmodel
AT weilin pedestriandetectionalgorithmintegratinglargekernelattentionandyolov5lightweightmodel
AT gengchao pedestriandetectionalgorithmintegratinglargekernelattentionandyolov5lightweightmodel
AT ranhaoxiang pedestriandetectionalgorithmintegratinglargekernelattentionandyolov5lightweightmodel
AT zhuhaodong pedestriandetectionalgorithmintegratinglargekernelattentionandyolov5lightweightmodel