Added link to checkpoints.

PiperOrigin-RevId: 579435387
google-research · Nov 4, 2023 · bf5a824 · bf5a824
1 parent b7da62b
commit bf5a824
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git 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
@@ -62,4 +62,8 @@ Set `<output_path>` in `jrdb_preprocess_train.py` and `jrdb_preprocess_test.py`
 
 ```python jrdb_preprocess_train.py```
 
-```python jrdb_preprocess_test.py```
+```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.