Mercurial > hg > nsaunier > traffic-intelligence

comparison scripts/classify-objects.py @ 685:94b291a5f933 dev

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
several updates for display
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 05 Jun 2015 17:13:28 +0200
parents 6d2ece4aed7c
children cdee6a3a47b4
comparison
equal deleted inserted replaced
684:6d2ece4aed7c 685:94b291a5f933
10 10
11 # TODO add mode detection live 11 # TODO add mode detection live
12 12
13 parser = argparse.ArgumentParser(description='The program processes indicators for all pairs of road users in the scene') 13 parser = argparse.ArgumentParser(description='The program processes indicators for all pairs of road users in the scene')
14 parser.add_argument('--cfg', dest = 'configFilename', help = 'name of the configuration file', required = True) 14 parser.add_argument('--cfg', dest = 'configFilename', help = 'name of the configuration file', required = True)
15 parser.add_argument('-n', dest = 'nObjects', help = 'number of objects to classify', type = int, default = None)
15 parser.add_argument('--plot-speed-distributions', dest = 'plotSpeedDistribution', help = 'simply plots the distributions used for each user type', action = 'store_true') 16 parser.add_argument('--plot-speed-distributions', dest = 'plotSpeedDistribution', help = 'simply plots the distributions used for each user type', action = 'store_true')
16 parser.add_argument('--max-speed-distribution-plot', dest = 'maxSpeedDistributionPlot', help = 'if plotting the user distributions, the maximum speed to display', type = float, default = 50.) 17 parser.add_argument('--max-speed-distribution-plot', dest = 'maxSpeedDistributionPlot', help = 'if plotting the user distributions, the maximum speed to display', type = float, default = 50.)
17 #parser.add_argument('-u', dest = 'undistort', help = 'undistort the video (because features have been extracted that way)', action = 'store_true') 18 #parser.add_argument('-u', dest = 'undistort', help = 'undistort the video (because features have been extracted that way)', action = 'store_true')
18 #parser.add_argument('-f', dest = 'firstFrameNum', help = 'number of first frame number to display', type = int) 19 #parser.add_argument('-f', dest = 'firstFrameNum', help = 'number of first frame number to display', type = int)
19 #parser.add_argument('--last-frame', dest = 'lastFrameNum', help = 'number of last frame number to save (for image saving, no display is made)', type = int) 20 #parser.add_argument('--last-frame', dest = 'lastFrameNum', help = 'number of last frame number to save (for image saving, no display is made)', type = int)
58 plt.legend() 59 plt.legend()
59 plt.title('Probability Density Function') 60 plt.title('Probability Density Function')
60 plt.show() 61 plt.show()
61 sys.exit() 62 sys.exit()
62 63
63 objects = storage.loadTrajectoriesFromSqlite(params.databaseFilename, 'object') 64 objects = storage.loadTrajectoriesFromSqlite(params.databaseFilename, 'object', args.nObjects, withFeatures = True)
64 features = storage.loadTrajectoriesFromSqlite(params.databaseFilename, 'feature') 65 #features = storage.loadTrajectoriesFromSqlite(params.databaseFilename, 'feature')
65 intervals = [] 66 intervals = []
66 for obj in objects: 67 for obj in objects:
67 obj.setFeatures(features) 68 #obj.setFeatures(features)
68 intervals.append(obj.getTimeInterval()) 69 intervals.append(obj.getTimeInterval())
69 timeInterval = moving.unionIntervals(intervals) 70 timeInterval = moving.unionIntervals(intervals)
70 71
71 capture = cv2.VideoCapture(params.videoFilename) 72 capture = cv2.VideoCapture(params.videoFilename)
72 width = int(capture.get(cv2.cv.CV_CAP_PROP_FRAME_WIDTH)) 73 width = int(capture.get(cv2.cv.CV_CAP_PROP_FRAME_WIDTH))