Trajnet tools.
Project description
Tools
summary table and plots: python -m trajnettools.summarize <dataset_files>
plot trajectories in a scene: python -m trajnettools.trajectories <dataset_file>
visualize interactions: python -m trajnettools.visualize_interactions <dataset_file> --interaction_type 'ca'
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 average_l2(), final_l2() and collision() functions
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
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 trajnettools-0.2.0-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3ec675852745a2b22a79b0454c5b80943b1edb6bc9111bbc95a3914d5f3bba55 |
|
MD5 | 136ae3c39f9710d7e288cc241dac2d32 |
|
BLAKE2b-256 | 592217ae54a6adcf708ddfe28b4bcff5d7d481beb05b58f13baffe6dc426469e |