Pedestrian Detection and Tracking Algorithm Based on GhostNet and Attention Mechanism
CSTR:
Author:
Affiliation:

Engineering Research Center for Metallurgical Automation and Measurement Technology of Ministry of Education, Wuhan University of Science and Technology, Wuhan 430081,China

Clc Number:

TP391.41

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Aiming at the problems of low accuracy and slow speed when only relying on traditional object detection and tracking algorithms in complex scenes, a pedestrian detection and tracking algorithm based on GhostNet and attention mechanism is proposed. First, the backbone network of YOLOv3 is replaced with GhostNet to retain the multi-scale prediction part, the Ghost module is used to reduce the parameters and calculations of the deep network model, and the attention mechanism is integrated into the Ghost module to give higher weight to important features. Then, the direct evaluation index GIoU of object detection is introduced to guide the regression task. Finally, the Deep-Sort algorithm is used for tracking. Experiments on public data sets show that: The mean Average precision (mAP) of the improved model reaches 92.53%, and the frame rate is 2.5 times that of the YOLOv3 model; The tracking accuracy of the proposed algorithm is better than that before the improvement and that of other algorithms; The algorithm can track multi-object pedestrians in complex scenes accurately and effectively, and has strong robustness.

    Reference
    Related
    Cited by
Get Citation

WANG Lihui, YANG Xianzhao, LIU Huikang, HUANG Jingjing. Pedestrian Detection and Tracking Algorithm Based on GhostNet and Attention Mechanism[J].,2022,37(1):108-121.

Copy
Related Videos

Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:January 23,2021
  • Revised:May 08,2021
  • Adopted:
  • Online: January 25,2022
  • Published:
Article QR Code