Skip to content

Commit

Permalink
add to Readme
Browse files Browse the repository at this point in the history
  • Loading branch information
stepankonev committed Nov 20, 2024
1 parent 5e3f06c commit b4fc1f8
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -116,6 +116,7 @@ Currently, the dataloader supports interfacing with the following datasets:
| UCY - Zara1 | `eupeds_zara1` | `train`, `val`, `train_loo`, `val_loo`, `test_loo` | `cyprus` | The Zara1 scene from the UCY Pedestrians dataset | 0.4s (2.5Hz) | |
| UCY - Zara2 | `eupeds_zara2` | `train`, `val`, `train_loo`, `val_loo`, `test_loo` | `cyprus` | The Zara2 scene from the UCY Pedestrians dataset | 0.4s (2.5Hz) | |
| Stanford Drone Dataset | `sdd` | `train`, `val`, `test` | `stanford` | Stanford Drone Dataset (60 scenes, randomly split 42/9/9 (70%/15%/15%) for training/validation/test) | 0.0333...s (30Hz) | |
| highD | `highD` | `all` | N/A | Traffic recordings for more than 110 500 vehicles | 0.04s (25Hz) | :white_check_mark: |

### Adding New Datasets
The code that interfaces the original datasets (dealing with their unique formats) can be found in `src/trajdata/dataset_specific`.
Expand Down

0 comments on commit b4fc1f8

Please sign in to comment.