PyTorch-Based Implementation of Label-Aware Graph Representation for Multi-Class Trajectory Prediction

Qianhui Men and Hubert P. H. Shum
Software Impacts (SIMPAC), 2021

 Impact Factor: 2.1

PyTorch-Based Implementation of Label-Aware Graph Representation for Multi-Class Trajectory Prediction

Abstract

Trajectory Prediction under diverse patterns has attracted increasing attention in multiple real-world applications ranging from urban traffic analysis to human motion understanding, among which graph convolution network (GCN) is frequently adopted with its superior ability in modeling the complex trajectory interactions among multiple humans. In this work, we propose a python package by enhancing GCN with class label information of the trajectory, such that we can explicitly model not only human trajectories but also that of other road users such as vehicles. This is done by integrating a label-embedded graph with the existing graph structure in the standard graph convolution layer. The flexibility and the portability of the package also allow researchers to employ it under more general multi-class sequential prediction tasks.

Downloads

YouTube

Citations

BibTeX

@article{men21pytorch,
 author={Men, Qianhui and Shum, Hubert P. H.},
 journal={Software Impacts},
 title={PyTorch-Based Implementation of Label-Aware Graph Representation for Multi-Class Trajectory Prediction},
 year={2021},
 volume={11},
 pages={100201},
 numpages={3},
 doi={10.1016/j.simpa.2021.100201},
 issn={2665-9638},
 publisher={Elsevier},
}

RIS

TY  - JOUR
AU  - Men, Qianhui
AU  - Shum, Hubert P. H.
T2  - Software Impacts
TI  - PyTorch-Based Implementation of Label-Aware Graph Representation for Multi-Class Trajectory Prediction
PY  - 2021
VL  - 11
SP  - 100201
EP  - 100201
DO  - 10.1016/j.simpa.2021.100201
SN  - 2665-9638
PB  - Elsevier
ER  - 

Plain Text

Qianhui Men and Hubert P. H. Shum, "PyTorch-Based Implementation of Label-Aware Graph Representation for Multi-Class Trajectory Prediction," Software Impacts, vol. 11, pp. 100201, Elsevier, 2021.

Supporting Grants

Similar Research

Ben Rainbow, Qianhui Men and Hubert P. H. Shum, "Semantics-STGCNN: A Semantics-Guided Spatial-Temporal Graph Convolutional Network for Multi-Class Trajectory Prediction", Proceedings of the 2021 IEEE International Conference on Systems, Man, and Cybernetics (SMC), 2021
Ruochen Li, Stamos Katsigiannis and Hubert P. H. Shum, "Multiclass-SGCN: Sparse Graph-Based Trajectory Prediction with Agent Class Embedding", Proceedings of the 2022 IEEE International Conference on Image Processing (ICIP), 2022

 

 

Last updated on 14 April 2024
RSS Feed