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...
Autores principales: | , , , , , |
---|---|
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 |