diff --git a/README.md b/README.md index 3f46cd4..5bc046f 100644 --- a/README.md +++ b/README.md @@ -132,7 +132,7 @@ On the ETH/UCY Pedestrians Dataset: The train / test split is implemented [here](/human_scene_transformer/config/jrdb/dataset_params.py). ### Checkpoints -You can download trained model checkpoints for both `JRDB` and `Pedestrians (ETH/UCY)` datasets [here]() (Coming Soon). +You can download trained model checkpoints for both `JRDB` and `Pedestrians (ETH/UCY)` datasets [here](https://storage.googleapis.com/gresearch/human_scene_transformer/checkpoints.zip). To evaluate the pre-trained checkpoints you will have to adjust the path to the dataset in the respective `params/operative_config.gin` file. diff --git a/human_scene_transformer/data/README.md b/human_scene_transformer/data/README.md index 8b697a5..9748c18 100644 --- a/human_scene_transformer/data/README.md +++ b/human_scene_transformer/data/README.md @@ -62,4 +62,8 @@ Set `` in `jrdb_preprocess_train.py` and `jrdb_preprocess_test.py` ```python jrdb_preprocess_train.py``` -```python jrdb_preprocess_test.py``` \ No newline at end of file +```python jrdb_preprocess_test.py``` + +Please note that this can take multiple hours due to the processing of the scene's +pointclouds. If you do not need the pointclouds you can speed up the processing +by setting `POINTCLOUD=False` in both files. \ No newline at end of file