Skip to main content
Join the official 2019 Python Developers SurveyStart the survey!

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:

docs/train/biwi_hotel.ndjson.theta.png docs/train/biwi_hotel.ndjson.speed.png

crowds_students001:

docs/train/crowds_students001.ndjson.theta.png docs/train/crowds_students001.ndjson.speed.png

crowds_students003:

docs/train/crowds_students003.ndjson.theta.png docs/train/crowds_students003.ndjson.speed.png

crowds_zara02:

docs/train/crowds_zara02.ndjson.theta.png docs/train/crowds_zara02.ndjson.speed.png

crowds_zara03:

docs/train/crowds_zara03.ndjson.theta.png docs/train/crowds_zara03.ndjson.speed.png

dukemtmc:

docs/train/dukemtmc.ndjson.theta.png docs/train/dukemtmc.ndjson.speed.png

syi:

docs/train/syi.ndjson.theta.png docs/train/syi.ndjson.speed.png

wildtrack:

docs/train/wildtrack.ndjson.theta.png docs/train/wildtrack.ndjson.speed.png

Interactions

leader_follower:

docs/train/crowds_zara02.ndjson_1_9.png docs/train/crowds_zara02.ndjson_1_9_full.png

collision_avoidance:

docs/train/crowds_zara02.ndjson_2_25.png docs/train/crowds_zara02.ndjson_2_25_full.png

group:

docs/train/crowds_zara02.ndjson_3_9.png docs/train/crowds_zara02.ndjson_3_9_full.png

others:

docs/train/crowds_zara02.ndjson_4_13.png docs/train/crowds_zara02.ndjson_4_13_full.png

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for trajnettools, version 0.2.0
Filename, size File type Python version Upload date Hashes
Filename, size trajnettools-0.2.0-py2.py3-none-any.whl (16.1 kB) File type Wheel Python version py2.py3 Upload date Hashes View hashes
Filename, size trajnettools-0.2.0.tar.gz (12.0 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page