自動運転車は、密集した不均一な交通に安全かつ合理的に参加するために、周囲の交通エージェントの動きのパターンを十分に分析し、将来の軌道を正確に予測する必要があります。トラフィックエージェントの軌道は、トラフィックエージェント自体だけでなく、相互の空間的相互作用によっても影響を受けるため、これは困難です。以前の方法は通常、長短期記憶ネットワーク(LSTM)の順次の段階的な処理に依存しており、単一タイプのトラフィックエージェントの空間ネイバー間の相互作用を抽出するだけです。時空間トランスフォーマーによって時空間相互作用をモデル化し、時間トランスフォーマーによって時間シーケンスを処理する時空間トランスフォーマーネットワーク(S2TNet)を提案します。トラフィックエージェントの異質性を処理するために、追加のカテゴリ、形状、および見出し情報をネットワークに入力します。提案された方法は、平均変位誤差と最終変位誤差の加重和の両方で、ApolloScape Trajectoryデータセットの最新の方法よりも7%以上優れています。私たちのコードはhttps://github.com/chenghuang66/s2tnetで入手できます。
To safely and rationally participate in dense and heterogeneous traffic, autonomous vehicles require to sufficiently analyze the motion patterns of surrounding traffic-agents and accurately predict their future trajectories. This is challenging because the trajectories of traffic-agents are not only influenced by the traffic-agents themselves but also by spatial interaction with each other. Previous methods usually rely on the sequential step-by-step processing of Long Short-Term Memory networks (LSTMs) and merely extract the interactions between spatial neighbors for single type traffic-agents. We propose the Spatio-Temporal Transformer Networks (S2TNet), which models the spatio-temporal interactions by spatio-temporal Transformer and deals with the temporel sequences by temporal Transformer. We input additional category, shape and heading information into our networks to handle the heterogeneity of traffic-agents. The proposed methods outperforms state-of-the-art methods on ApolloScape Trajectory dataset by more than 7% on both the weighted sum of Average and Final Displacement Error. Our code is available at https://github.com/chenghuang66/s2tnet.