Mercurial > hg > nsaunier > traffic-intelligence

annotate scripts/replay-event-annotation.py @ 759:a05b70f307dd dev

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added function to count vehicles per VISSIM link
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 04 Nov 2015 17:12:06 -0500
parents ba813f148ade
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
381
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
1 #! /usr/bin/env python
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
2
382
ba813f148ade development for clustering
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 381
diff changeset
3 import sys, argparse, datetime
381
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
4
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
5 import storage, cvutils, utils
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
6
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
7 import matplotlib.pylab as pylab
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
8 import matplotlib.pyplot as plt
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
9 import numpy as np
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
10
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
11
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
12 annotations = pylab.csv2rec(sys.argv[1])
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
13
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
14 frameRate = 30
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
15 dirname = "/home/nicolas/Research/Data/montreal/infractions-pietons/"
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
16 videoDirnames = {'amherst': '2011-06-22-sherbrooke-amherst/',
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
17 'iberville': '2011-06-28-sherbrooke-iberville/'}
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
18
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
19 # for amherst, subtract 40 seconds: add a delta
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
20
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
21 for annotation in annotations:
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
22 video = annotation['video_name'].lower()
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
23 print('{} {}'.format(annotation['conflict_start_time'], annotation['conflict_end_time']))
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
24 print(annotation['road_user_1']+' '+annotation['road_user_2']+' '+annotation['conflict_quality'])
387cc0142211 script to replay event annotations
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
25 print(annotation['comments'])
382
ba813f148ade development for clustering
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 381
diff changeset
26 cvutils.playVideo(dirname+videoDirnames[video]+video+'-{}.avi'.format(annotation['video_start_time']), utils.timeToFrames(annotation['conflict_start_time']+datetime.timedelta(seconds=-40), frameRate), frameRate, True, False, annotation['road_user_1']+' '+annotation['road_user_2']+' '+annotation['conflict_quality'])