Henry1iu / TNT-Trajectory-Prediction

A Pytorch Implementation of TNT: Target-driveN Trajectory Prediction
487 stars 92 forks source link

Representation of vector node (feature) #18

Closed legendjack closed 2 years ago

legendjack commented 2 years ago

Hi,感谢分享的代码,我注意到你使用点和向量表示折线的向量特征,而在yet_another_vectornet和VectorNet论文中(公式1)都是使用的start_point和end_point表示的向量特征;请问您使用点和向量是基于什么原因呢?

legendjack commented 2 years ago

另外我在train的时候发现Target_Offset_Loss一直在震荡,不下降,请问你遇到过这个问题吗?

Henry1iu commented 2 years ago

Hi,

关于输入的representation, 我认为我使用的representation与原作的区别不大 (只是一个linear transformation).

Offset loss震荡的问题现在还有, 有可能是postive example太少的问题, 我还在调整模型试图解决, 请关注后续更新.

Best, Jianbang