基于时空图注意力网络的车辆多模态轨迹预测模型

陈文强,王东丹,朱文英,汪勇杰,王涛

Vehicle multimodal trajectory prediction model based on spatio-temporal graph attention network

Wenqiang CHEN,Dongdan WANG,Wenying ZHU,Yongjie WANG,Tao WANG

表 2 模型每轮训练时间对比

Tab.2 Comparison of training time per epoch for models

模型名称 te/s S-LSTM[16] 0.0157 CS-LSTM[15] 0.0174 S-GAN[14] 0.0283 PIP[18] 0.0185 STDAN[17] 0.0214 STGAMT(w/o IFE) 0.0137 STGAMT 0.0139