Vehicle Trajectory Prediction 自用,记录课题进展
只有LSTM进行预测的程序,9个点预测一个点
主程序my0224
读取模型进行测试的程序Modeltest
attention-based seq2seq模型,自己随便写的,没有进行参数优化
使用了inD数据集,数据处理位于data_process文件夹
由于某些原因,不提供任何与数据集相关的文件,需要数据集请去inD官网申请
使用attention-based-seq2seq0308.py进行训练,Test.py进行测试
使用前需要修改attention-based-seq2seq0308.py中导入dataprocess方法的路径
该仓库不再上传代码,仅作记录 上传了MyVectorNet的代码