A Spatio-temporal Graph Network Allowing Incomplete Trajectory Input for Pedestrian Trajectory Prediction

摘要

Pedestrian trajectory prediction is important in the research of mobile robot navigation in environments with pedestrians. Most pedestrian trajectory prediction algorithms require as input complete historical trajectories. If a pedestrian is unobservable in any frame in the past, then its historical trajectory becomes incomplete and the algorithm does not predict its future trajectory. To address this limitation, we propose STGN-IT, a spatio-temporal graph network allowing incomplete trajectory input. STGN-IT is able to predict the future trajectories of pedestrians with incomplete historical trajectories. STGN-IT uses the spatio-temporal graph with an additional encoding method to represent the historical trajectories and observation states of pedestrians. Moreover, STGN-IT introduces static obstacles in the environment that may affect the future trajectories as nodes to further improve the prediction accuracy. A clustering algorithm is also applied in the construction of spatio-temporal graphs. Experiments on public datasets show that STGN-IT outperforms state-of-the-art algorithms. Code will be released upon publication.