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 条
  • [21] PTP-STGCN: Pedestrian Trajectory Prediction Based on a Spatio-temporal Graph Convolutional Neural Network
    Lian, Jing
    Ren, Weiwei
    Li, Linhui
    Zhou, Yafu
    Zhou, Bin
    [J]. APPLIED INTELLIGENCE, 2023, 53 (03) : 2862 - 2878
  • [22] Multi-scale Dilated Convolutional Neural Network Model Based on Attention Mechanism
    Wang, Jingbin
    Lai, Xiaolian
    Lei, Jing
    Zhang, Jingxuan
    [J]. Moshi Shibie yu Rengong Zhineng/Pattern Recognition and Artificial Intelligence, 2021, 34 (06): : 497 - 508
  • [23] Instance segmentation convolutional neural network based on multi-scale attention mechanism
    Wang Gaihua
    Lin Jinheng
    Cheng Lei
    Dai Yingying
    Zhang Tianlun
    [J]. PLOS ONE, 2022, 17 (01):
  • [24] Atten-GAN: Pedestrian Trajectory Prediction with GAN Based on Attention Mechanism
    Fang, Fang
    Zhang, Pengpeng
    Zhou, Bo
    Qian, Kun
    Gan, Yahui
    [J]. COGNITIVE COMPUTATION, 2022, 14 (06) : 2296 - 2305
  • [25] Pedestrian Reidentification Algorithm Based on Deconvolution Network Feature Extraction-Multilayer Attention Mechanism Convolutional Neural Network
    An, Feng-Ping
    Liu, Jun-e
    Bai, Lei
    [J]. JOURNAL OF SENSORS, 2021, 2021
  • [26] Atten-GAN: Pedestrian Trajectory Prediction with GAN Based on Attention Mechanism
    Fang Fang
    Pengpeng Zhang
    Bo Zhou
    Kun Qian
    Yahui Gan
    [J]. Cognitive Computation, 2022, 14 : 2296 - 2305
  • [27] MRCNNAM: Mask Region Convolutional Neural Network Model Based On Attention Mechanism And Gabor Feature For Pedestrian Detection
    Wang, Ye
    [J]. JOURNAL OF APPLIED SCIENCE AND ENGINEERING, 2023, 26 (11): : 1555 - 1561
  • [28] Drug Response Prediction Based on 1D Convolutional Neural Network and Attention Mechanism
    Zhu, Mingxun
    Meng, Zhigang
    Wang, Jingyi
    [J]. COMPUTATIONAL AND MATHEMATICAL METHODS IN MEDICINE, 2022, 2022
  • [29] STM-GCN: a spatiotemporal multi-graph convolutional network for pedestrian trajectory prediction
    Taki Youssef
    Elmoukhtar Zemmouri
    Anas Bouzid
    [J]. The Journal of Supercomputing, 2023, 79 : 20923 - 20937
  • [30] PTPGC: Pedestrian trajectory prediction by graph attention network with ConvLSTM
    Yang, Juan
    Sun, Xu
    Wang, Rong Gui
    Xue, Li Xia
    [J]. ROBOTICS AND AUTONOMOUS SYSTEMS, 2022, 148