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'])