Trajnet++ tools.
Project description
Tools
summary table and plots: python -m trajnettools.summarize <dataset_files>
plot sample trajectories: python -m trajnettools.trajectories <dataset_file>
visualize interactions: python -m trajnettools.visualize_type <dataset_file>
obtain distribution of trajectory types: python -m trajnettools.dataset_stats <dataset_file>
APIs
trajnettools.Reader: class to read the dataset_file
trajnettools.show: module containing contexts for visualizing rows and paths
trajnettools.writers: write a trajnet dataset file
trajnettools.metrics: contains unimodal metrics: average_l2(), final_l2() and collision() and multimodal metrics: topk() and nll() implementations
Dataset
Datasets are split into train, val and test set. Every line is a self contained JSON string (ndJSON).
Scene:
{"scene": {"id": 266, "p": 254, "s": 10238, "e": 10358, "fps": 2.5, "tag": 2}}
Track:
{"track": {"f": 10238, "p": 248, "x": 13.2, "y": 5.85}}
with:
id: scene id
p: pedestrian id
s, e: start and end frame id
fps: frame rate
tag: trajectory type
f: frame id
x, y: x- and y-coordinate in meters
pred_number: (optional) prediction number for multiple output predictions
scene_id: (optional) corresponding scene_id for multiple output predictions
Frame numbers are not recomputed. Rows are resampled to about 2.5 rows per second.
Dev
pylint trajnettools
python -m pytest
# optional: mypy trajnettools --disallow-untyped-defs
Dataset Summaries
biwi_hotel:
crowds_students001:
crowds_students003:
crowds_zara02:
crowds_zara03:
dukemtmc:
syi:
wildtrack:
Interactions
leader_follower:
collision_avoidance:
group:
others:
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for trajnetplusplustools-0.3.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | ef6773c99f1ec7cb2480ef9f96c6d63a4e2d644aa1ba0f658ba5f23c7833f6f3 |
|
MD5 | 4117cc360026e77d9b0e865b677879a5 |
|
BLAKE2b-256 | 35d82e36edd8b4d2cc316cfab7acb70bdca4f4b8f830323eb5191397cd8c2a67 |
Hashes for trajnetplusplustools-0.3.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 79ff367ab34bdd0504e8e22d2f27f937e0f601388bd6a1546faac4e2bfa19708 |
|
MD5 | bfcc2f6126b1687d152c08c8607c091a |
|
BLAKE2b-256 | 560001a99c1e9e5c63ff6219a94bb9ebe0a6586481a1b71cae1d9eea6525c657 |