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