Skip to content

Code of our work "Maneuver-based Anchor Trajectory Hypotheses at Roundabouts".

License

Notifications You must be signed in to change notification settings

m-hasan-n/roundabout

Repository files navigation

Maneuver-based Anchor Trajectory Hypotheses at Roundabouts

We extend existing recurrent encoder-decoder models to be advantageously combined with anchor trajectories to predict vehicle behaviors on a roundabout. Drivers’ intentions are encoded by a set of maneuvers that correspond to semantic driving concepts. Accordingly, our model employs a set of maneuver-specific anchor trajectories that cover the space of possible outcomes at the roundabout. The proposed model can output a multi- modal distribution over the predicted future trajectories based on the maneuver-specific anchors. We evaluate our model using the public RounD dataset.

model image

RounD Dataset Pre-processing

The RounD Dataset is a new dataset of naturalistic road user trajectories recorded at German roundabouts. Download the dataset to the 'rounD' directory, then run the following MATLAB script:

preprocess_rounD.m

This will do the required pre-processing, split the dataset into train, validation and test subsets, and save such subsets into the 'data' directory.

Model Arguments

The default network arguments are in:

model_args.py 

You can set the required experiment arguments in this script. For example:

  • args['ip_dim'] selects the input dimensionality (2D or 3D).

  • args['use_intention'] and args['use_anchors'] are Boolean variables that choose whether using intention prediction and anchor trajectories or not.

Model Training and Evaluation

The model structure is coded in 'model.py'. After setting the required experiment arguments, create a 'trained_models' directory to save the trained models. You can start model training by running:

train.py

To test a trained model, first create an 'eval_res' directory, then run:

evaluate.py

which will load and test the trained model defined by the selected model arguments. The RMSE results will be saved as csv files to the 'eval_res' directory.

Citation

If you find this code useful for your research, please cite our work:

  • Mohamed Hasan, Evangelos Paschalidis, Albert Solernou, He Wang, Gustav Markkula and Richard Romano, "Maneuver-based Anchor Trajectory Hypotheses at Roundabouts", preprint 2021.

License

This project is licensed under the MIT License - see the LICENSE.md file for details.

Releases

No releases published

Packages

No packages published