Multi-information-based convolutional neural network with attention mechanism for pedestrian trajectory prediction

被引:14
|
作者
Wang, Ruiping [1 ]
Cui, Yong [1 ]
Song, Xiao [2 ]
Chen, Kai [1 ]
Fang, Hong [3 ]
机构
[1] Beihang Univ BUAA, Sch Automat Sci & Elect Engn, Beijing, Peoples R China
[2] Beihang Univ BUAA, Sch Cyber Sci & Technol, Beijing, Peoples R China
[3] Sci & Technol Space Phys Lab, Beijing, Peoples R China
关键词
Depth map; Pose; 2D-3D size information; Convolutional neural network; Trajectory prediction; MODELS;
D O I
10.1016/j.imavis.2021.104110
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Predicting pedestrian trajectory is useful in many applications, such as autonomous driving and unmanned vehicles. However, it is a challenging task because of the complexity of the interactions among pedestrians and the environment. Most existing works employ long short-term memory networks to learn pedestrian behaviors, but their prediction accuracy is not good, and their computing speed is relatively slow. To tackle this problem, we propose a multi-information-based convolutional neural network (MI-CNN) to incorporate the historical trajectory, depth map, pose, and 2D-3D size information to predict the future trajectory of the pedestrian subject. After training, we evaluate our model on crowded videos in the public datasets MOT16 and MOT20. Experiments demonstrate that the proposed method outperforms state-of-the-art approaches both in prediction accuracy and computing speed. (c) 2021 Published by Elsevier B.V.
引用
收藏
页数:10
相关论文
共 50 条
  • [1] Pedestrian Trajectory Prediction Based on GAN and Attention Mechanism
    Ouyang Jun
    Shi Qingwei
    Wang Xinxin
    Wang Liang
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2020, 57 (14)
  • [2] Pedestrian trajectory prediction with convolutional neural networks
    Zamboni, Simone
    Kefato, Zekarias Tilahun
    Girdzijauskas, Sarunas
    Noren, Christoffer
    Dal Col, Laura
    [J]. PATTERN RECOGNITION, 2022, 121
  • [3] Pedestrian Trajectory Prediction Based on Deep Convolutional LSTM Network
    Song, Xiao
    Chen, Kai
    Li, Xu
    Sun, Jinghan
    Hou, Baocun
    Cui, Yong
    Zhang, Baochang
    Xiong, Gang
    Wang, Zilie
    [J]. IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2021, 22 (06) : 3285 - 3302
  • [4] Pedestrian trajectory prediction algorithm based on graph convolutional network
    Wang, Tianbao
    Liu, Yu
    Guo, Jichang
    Jin, Weipei
    [J]. Harbin Gongye Daxue Xuebao/Journal of Harbin Institute of Technology, 2021, 53 (02): : 53 - 60
  • [5] Multi Attention Generative Adversarial Network for Pedestrian Trajectory Prediction Based on Spatial Gridding
    An, Huihui
    Liu, Miao
    Wang, Xiaolan
    Zhang, Weiwei
    Gong, Jun
    [J]. AUTOMOTIVE INNOVATION, 2024, 7 (03) : 443 - 455
  • [6] MSTCNN: multi-modal spatio-temporal convolutional neural network for pedestrian trajectory prediction
    Haifeng Sang
    Wangxing Chen
    Haifeng Wang
    Jinyu Wang
    [J]. Multimedia Tools and Applications, 2024, 83 : 8533 - 8550
  • [7] MSTCNN: multi-modal spatio-temporal convolutional neural network for pedestrian trajectory prediction
    Sang, Haifeng
    Chen, Wangxing
    Wang, Haifeng
    Wang, Jinyu
    [J]. MULTIMEDIA TOOLS AND APPLICATIONS, 2024, 83 (03) : 8533 - 8550
  • [8] Fully Convolutional Encoder-Decoder With an Attention Mechanism for Practical Pedestrian Trajectory Prediction
    Chen, Kai
    Song, Xiao
    Yuan, Haitao
    Ren, Xiaoxiang
    [J]. IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (11) : 20046 - 20060
  • [9] Compact Convolutional Neural Network with Multi-Headed Attention Mechanism for Seizure Prediction
    Ding, Xin
    Nie, Weiwei
    Liu, Xinyu
    Wang, Xiuying
    Yuan, Qi
    [J]. INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2023, 33 (03)
  • [10] Disentangled Multi-Relational Graph Convolutional Network for Pedestrian Trajectory Prediction
    Bae, Inhwan
    Jeon, Hae-Gon
    [J]. THIRTY-FIFTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THIRTY-THIRD CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE AND THE ELEVENTH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2021, 35 : 911 - 919