Official codes the for paper: An Ensemble Learning Framework for Trajectory Prediction in Interactive Scenarios
These codes are official codes for paper An Ensemble Learning Framework for Trajectory Prediction in Interactive Scenarios, Zirui Li, Yunlong Lin et al., Beijing Institute of Technology. The framework in this paper is termed Interactive Ensemble Trajectory Predictor (IETP).
We provides codes of our experiments in the paper. It is an example of IETP which uses CSP models as base learners. The experiments in the paper are based on NGSIM datasets.
If you want to run the codes, please download the NGSIM datasets and create suitable directories in your own file folders. The data-processing in this case is the same as it in the paper: convolutional social pooling for vehicle trajectory prediction.
If using this code, parts of it, or developments from it, please cite our paper:
@INPROCEEDINGS{9827070, author={Li, Zirui and Lin, Yunlong and Gong, Cheng and Wang, Xinwei and Liu, Qi and Gong, Jianwei and Lu, Chao}, booktitle={2022 IEEE Intelligent Vehicles Symposium (IV)}, title={An Ensemble Learning Framework for Vehicle Trajectory Prediction in Interactive Scenarios}, year={2022}, volume={}, number={}, pages={51-57}, doi={10.1109/IV51971.2022.9827070}}