diff samples/TTC/ttcCompare.py @ 464:dcc821b98efc

integrated and reorganized Sohail s work on exact ttc computation
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Sun, 23 Feb 2014 23:18:08 -0500
parents TTC Sample/ttcCompare.py@af2222c0c9c0
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/samples/TTC/ttcCompare.py	Sun Feb 23 23:18:08 2014 -0500
@@ -0,0 +1,51 @@
+import matplotlib.image as mpimg
+from numpy.linalg.linalg import inv
+from numpy import loadtxt
+import computeTTC
+import numpy as np
+import glob
+
+#databaseName = ['11.sqlite']
+databaseName = glob.glob("*.sqlite")
+
+bikOri = mpimg.imread('bikeOri.png')
+bikAreaOri = bikOri[:,:,0]
+bikDes = mpimg.imread('bikeDes.png')
+bikAreaDes = bikDes[:,:,0]
+carOri = mpimg.imread('carOri.png')
+carAreaOri = carOri[:,:,0]
+carDes = mpimg.imread('carDes.png')
+carAreaDes = carDes[:,:,0]
+
+videoY = bikOri.shape[0]
+videoX = bikOri.shape[1]
+
+homography=inv(loadtxt('homography.txt'))
+framePerSecond=15
+collisionDistanceThreshold=10
+
+bikCount=0
+carCount=0
+bikSpeed=[]
+carSpeed=[]
+ttcDist=[]
+potentioalCollision=0
+
+for d in databaseName:
+    print d
+    bc,cc,bs,cs,ttcdist,potCollision = computeTTC.computeTTC(d,homography,framePerSecond,videoX,videoY,collisionDistanceThreshold,bikAreaOri,bikAreaDes,carAreaOri,carAreaDes)
+    bikCount += bc
+    carCount += cc
+    bikSpeed.extend(bs)
+    carSpeed.extend(cs)
+    ttcDist.extend(ttcdist)
+    potentioalCollision += potCollision
+
+print 'bikCount :', bikCount
+print 'carCount :', carCount
+print 'Number of Potential Collisions :', potentioalCollision
+np.savetxt('ttc.txt',ttcDist,fmt='%1.2f')
+np.savetxt('freq.txt',[bikCount,carCount,potentioalCollision],fmt='%i')
+np.savetxt('bikSpeed.txt',bikSpeed,fmt='%1.2f')
+np.savetxt('carSpeed.txt',carSpeed,fmt='%1.2f')
+