Awesome

:trophy: Winner of the 2023 JRDB Trajectory Prediction Challenge - Reproduce our Result!

Human Scene Transformer

The (Human) Scene Transformer architecture (as described here and here) is a general and extendable trajectory prediction framework which threats trajectory prediction as a sequence to sequence problem and models it in a Transformer architecture.

It is straightforward to extend with

• additional input features
• custom environment encoder
• different loss functions
• ...

This is not an officially supported Google product.

Anticipating the motion of all humans in dynamic environments such as homes and offices is critical to enable safe and effective robot navigation. Such spaces remain challenging as humans do not follow strict rules of motion and there are often multiple occluded entry points such as corners and doors that create opportunities for sudden encounters. In this work, we present a Transformer based architecture to predict human future trajectories in human-centric environments from input features including human positions, head orientations, and 3D skeletal keypoints from onboard in-the-wild sensory information. The resulting model captures the inherent uncertainty for future human trajectory prediction and achieves state-of-the-art performance on common prediction benchmarks and a human tracking dataset captured from a mobile robot adapted for the prediction task. Furthermore, we identify new agents with limited historical data as a major contributor to error and demonstrate the complementary nature of 3D skeletal poses in reducing prediction error in such challenging scenarios.

If you use this work please cite our paper

@article{salzmann2023hst,
  title={Robots That Can See: Leveraging Human Pose for Trajectory Prediction},
  author={Salzmann, Tim and Chiang, Lewis and Ryll, Markus and Sadigh, Dorsa and Parada, Carolina and Bewley, Alex}
  journal={IEEE Robotics and Automation Letters},
  title={Robots That Can See: Leveraging Human Pose for Trajectory Prediction},
  year={2023}, volume={8}, number={11}, pages={7090-7097},
  doi={10.1109/LRA.2023.3312035}
}

Prerequisites

Install requirements via pip install -r requirements.txt.

Please note that this codebase is not compatible with the Intel MKL backend for tensorflow. The MKL backend supports tensors up to 5 dimensions which is not sufficient for parts of this codebase. Should you have a MKL backed tensorflow installation or are running into MKL related errors, please disable the tensorflow MKL backend by setting the environment variable TF_ENABLE_ONEDNN_OPTS=0 and TF_DISABLE_MKL=1.

Data

JRDB

We provide a extensive prep-processing pipeline to convert the JRDB dataset, JRDB was created as a detection and tracking dataset rather than a prediction dataset. To make the data suitable for a prediction task, we first extract the robot motion from the raw sensor data to account for the robot's motion. Further, on the JRDB training split we combine algorithmic detection with the ground truth labels from the tracking dataset to create authentic tracks as input and labels for HST. Note that we do not purely use the ground truth hand labeled tracks in the JRDB train dataset as we find them to be overly smoothed giving away the future human movement. To adapt the JRDB dataset for prediction please follow this README.

Make sure to adapt <data_path> in config/<jrdb/pedestrians>/dataset_params.gin accordingly.

If you want to use the JRDB dataset for trajectory prediction in PyTorch we provide a PyTorch Dataset wrapper for the processed dataset.

Pedestrians ETH/UCY

Please download the raw data here.

Training

JRDB

python train.py --model_base_dir=./model/jrdb  --gin_files=./config/jrdb/training_params.gin --gin_files=./config/jrdb/model_params.gin --gin_files=./config/jrdb/dataset_params.gin --gin_files=./config/jrdb/metrics.gin --dataset=JRDB

Pedestrians ETH/UCY

python train.py --model_base_dir=./models/pedestrians_eth  --gin_files=..config/pedestrians/training_params.gin --gin_files=..config/pedestrians/model_params.gin --gin_files=./config/pedestrians/dataset_params.gin --gin_files=./config/pedestrians/metrics.gin --dataset=PEDESTRIANS

JRDB Trajectory Prediction Challenge Results

To reproduce our winning results in the 2023 JRDB Trajectory Prediction Challenge:

• Make sure that you follow the data pre-processing instructions and pay special attention to where the instructions differentiate between the JRDB Challenge dataset and the original paper dataset.

• Download the trained challenge model here

• Run

python jrdb/eval_challenge.py --model_path=<path_to_challenge_model_folder> --checkpoint_path=<path_to_challenge_model_folder>/ckpts/ckpt-20 --dataset_path=<dataset_path> --output_path=<result_folder>

Evaluation

JRDB

python jrdb/eval.py --model_path=./models/jrdb/ --checkpoint_path=./models/jrdb/ckpts/ckpt-30

Keypoints Impact Evaluation

python jrdb/eval_keypoints.py --model_path=./models/jrdb/ --checkpoint_path=./models/jrdb/ckpts/ckpt-30

vs

python jrdb/eval_keypoints.py --model_path=./models/jrdb_no_keypoints/ --checkpoint_path=./models/jrdb_no_keypoints/ckpts/ckpt-30

Pedestrians ETH/UCY

python pedestrians/eval.py --model_path=./models/pedestrians_eth/ --checkpoint_path=./models/pedestrians_eth/ckpts/ckpt-20

Results

Compared to the published paper we improved our data processing and fixed small bugs in this code release. If you compare against our method please use the following updated results.

On the JRDB dataset with dataset options as set here:

On the ETH/UCY Pedestrians Dataset:

JRDB Train / Test Split

The train / test split is implemented here.

Checkpoints

You can download trained model checkpoints for both JRDB and Pedestrians (ETH/UCY) datasets here.

To evaluate the pre-trained checkpoints you will have to adjust the path to the dataset in the respective params/operative_config.gin file.

Runtime

Evaluation of forward inference runtime with single output mode: