Mercurial Hosting > traffic-intelligence
diff scripts/replay-event-annotation.py @ 381:387cc0142211
script to replay event annotations
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Fri, 19 Jul 2013 11:58:35 -0400 |
parents | |
children | ba813f148ade |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/scripts/replay-event-annotation.py Fri Jul 19 11:58:35 2013 -0400 @@ -0,0 +1,26 @@ +#! /usr/bin/env python + +import sys, argparse + +import storage, cvutils, utils + +import matplotlib.pylab as pylab +import matplotlib.pyplot as plt +import numpy as np + + +annotations = pylab.csv2rec(sys.argv[1]) + +frameRate = 30 +dirname = "/home/nicolas/Research/Data/montreal/infractions-pietons/" +videoDirnames = {'amherst': '2011-06-22-sherbrooke-amherst/', + 'iberville': '2011-06-28-sherbrooke-iberville/'} + +# for amherst, subtract 40 seconds: add a delta + +for annotation in annotations: + video = annotation['video_name'].lower() + print('{} {}'.format(annotation['conflict_start_time'], annotation['conflict_end_time'])) + print(annotation['road_user_1']+' '+annotation['road_user_2']+' '+annotation['conflict_quality']) + print(annotation['comments']) + cvutils.playVideo(dirname+videoDirnames[video]+video+'-{}.avi'.format(annotation['video_start_time']), utils.timeToFrames(annotation['conflict_start_time'], frameRate), frameRate, True, False, annotation['road_user_1']+' '+annotation['road_user_2']+' '+annotation['conflict_quality'])