GNN-Enhanced YOLOv4 for Pedestrian Detection

被引:0
作者
Peng, Furong [1 ]
Zhang, Aofeng [1 ]
Peng, Zhen [1 ]
机构
[1] Hunan Inst Engn, Coll Informat Sci & Engn, Xiangtan, Hunan, Peoples R China
来源
2024 3RD INTERNATIONAL CONFERENCE ON ROBOTICS, ARTIFICIAL INTELLIGENCE AND INTELLIGENT CONTROL, RAIIC 2024 | 2024年
关键词
Pedestrian detection; YOLOV4; Graph Neural Network;
D O I
10.1109/RAIIC61787.2024.10670892
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The YOLOv4 pedestrian detection method originally efficiently integrated multi-level feature maps through PAN networks, greatly improving the efficiency of information aggregation and transmission. However, the downsampling and upsampling operations in PAN networks during processing may cause information loss and introduce unnecessary noise during upsampling. To overcome this bottleneck, this paper proposes an innovative YOLOv4 pedestrian detection method, which introduces Graph Neural Network (GNN) instead of PAN network. Through GNN, this method can automatically optimize the information aggregation and transmission process, while effectively avoiding the potential problems of downsampling and upsampling, significantly improving the accuracy and efficiency of pedestrian detection. The experimental results show that our method achieved 89.3% AP value and 92.0% AR value on the VOC2007 dataset.
引用
收藏
页码:384 / 388
页数:5
相关论文
共 20 条
  • [1] Bochkovskiy A, 2020, Arxiv, DOI arXiv:2004.10934
  • [2] Chen X, 2023, INT C EL INF ENG COM, V2602, P844
  • [3] Dosovitskiy A, 2021, Arxiv, DOI arXiv:2010.11929
  • [4] Survey of Pedestrian Detection for Advanced Driver Assistance Systems
    Geronimo, David
    Lopez, Antonio M.
    Sappa, Angel D.
    Graf, Thorsten
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2010, 32 (07) : 1239 - 1258
  • [5] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [6] Rich feature hierarchies for accurate object detection and semantic segmentation
    Girshick, Ross
    Donahue, Jeff
    Darrell, Trevor
    Malik, Jitendra
    [J]. 2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, : 580 - 587
  • [7] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [8] Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2015, 37 (09) : 1904 - 1916
  • [9] PFYOLOv4: An Improved Small Object Pedestrian Detection Algorithm
    Li, Kaihui
    Zhuang, Yuan
    Lai, Jinling
    Zeng, Yunhui
    [J]. IEEE ACCESS, 2023, 11 : 17197 - 17206
  • [10] Redmon J., 2018, arXiv, DOI DOI 10.48550/ARXIV.1804.02767