Mercurial Hosting > traffic-intelligence

comparison TTC Sample/ttcCompare.py @ 462:af2222c0c9c0

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
TTC tested and updatet!
author Sohail Zangenehpour <sohail.zangenehpour@mail.mcgill.ca>
date Mon, 03 Feb 2014 15:41:57 -0500
parents
children
comparison
equal deleted inserted replaced
461:cb41f9a4652b 462:af2222c0c9c0
1 import matplotlib.image as mpimg
2 from numpy.linalg.linalg import inv
3 from numpy import loadtxt
4 import computeTTC
5 import numpy as np
6 import glob
7
8 #databaseName = ['11.sqlite']
9 databaseName = glob.glob("*.sqlite")
10
11 bikOri = mpimg.imread('bikeOri.png')
12 bikAreaOri = bikOri[:,:,0]
13 bikDes = mpimg.imread('bikeDes.png')
14 bikAreaDes = bikDes[:,:,0]
15 carOri = mpimg.imread('carOri.png')
16 carAreaOri = carOri[:,:,0]
17 carDes = mpimg.imread('carDes.png')
18 carAreaDes = carDes[:,:,0]
19
20 videoY = bikOri.shape[0]
21 videoX = bikOri.shape[1]
22
23 homography=inv(loadtxt('homography.txt'))
24 framePerSecond=15
25 collisionDistanceThreshold=10
26
27 bikCount=0
28 carCount=0
29 bikSpeed=[]
30 carSpeed=[]
31 ttcDist=[]
32 potentioalCollision=0
33
34 for d in databaseName:
35 print d
36 bc,cc,bs,cs,ttcdist,potCollision = computeTTC.computeTTC(d,homography,framePerSecond,videoX,videoY,collisionDistanceThreshold,bikAreaOri,bikAreaDes,carAreaOri,carAreaDes)
37 bikCount += bc
38 carCount += cc
39 bikSpeed.extend(bs)
40 carSpeed.extend(cs)
41 ttcDist.extend(ttcdist)
42 potentioalCollision += potCollision
43
44 print 'bikCount :', bikCount
45 print 'carCount :', carCount
46 print 'Number of Potential Collisions :', potentioalCollision
47 np.savetxt('ttc.txt',ttcDist,fmt='%1.2f')
48 np.savetxt('freq.txt',[bikCount,carCount,potentioalCollision],fmt='%i')
49 np.savetxt('bikSpeed.txt',bikSpeed,fmt='%1.2f')
50 np.savetxt('carSpeed.txt',carSpeed,fmt='%1.2f')
51