Mercurial Hosting > traffic-intelligence
comparison python/utils.py @ 364:a50a69e04c2a
script modification so that command line arguments take precedence over config file
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Fri, 12 Jul 2013 02:29:49 -0400 |
parents | 72aa44072093 |
children | 90bdabc06e9f |
comparison
equal
deleted
inserted
replaced
363:68861b52a319 | 364:a50a69e04c2a |
---|---|
439 | 439 |
440 class TrackingParameters: | 440 class TrackingParameters: |
441 def loadConfigFile(self, filename): | 441 def loadConfigFile(self, filename): |
442 from ConfigParser import ConfigParser | 442 from ConfigParser import ConfigParser |
443 from numpy import loadtxt | 443 from numpy import loadtxt |
444 | 444 from os import path |
445 | |
445 config = ConfigParser() | 446 config = ConfigParser() |
446 config.readfp(FakeSecHead(openCheck(filename))) | 447 config.readfp(FakeSecHead(openCheck(filename))) |
447 self.sectionHeader = config.sections()[0] | 448 self.sectionHeader = config.sections()[0] |
448 self.videoFilename = config.get(self.sectionHeader, 'video-filename') | 449 self.videoFilename = config.get(self.sectionHeader, 'video-filename') |
449 self.databaseFilename = config.get(self.sectionHeader, 'database-filename') | 450 self.databaseFilename = config.get(self.sectionHeader, 'database-filename') |
450 self.homographyFilename = config.get(self.sectionHeader, 'homography-filename') | 451 self.homographyFilename = config.get(self.sectionHeader, 'homography-filename') |
451 self.homography = loadtxt(self.homographyFilename) | 452 if (path.exists(self.homographyFilename)): |
453 self.homography = loadtxt(self.homographyFilename) | |
454 else: | |
455 self.homography = None | |
452 self.firstFrameNum = config.getint(self.sectionHeader, 'frame1') | 456 self.firstFrameNum = config.getint(self.sectionHeader, 'frame1') |
453 self.videoFrameRate = config.getfloat(self.sectionHeader, 'video-fps') | 457 self.videoFrameRate = config.getfloat(self.sectionHeader, 'video-fps') |
454 | 458 |
455 self.maxPredictedSpeed = config.getfloat(self.sectionHeader, 'max-predicted-speed')/3.6/self.videoFrameRate | 459 self.maxPredictedSpeed = config.getfloat(self.sectionHeader, 'max-predicted-speed')/3.6/self.videoFrameRate |
456 self.predictionTimeHorizon = config.getfloat(self.sectionHeader, 'prediction-time-horizon')*self.videoFrameRate | 460 self.predictionTimeHorizon = config.getfloat(self.sectionHeader, 'prediction-time-horizon')*self.videoFrameRate |