Mercurial Hosting > traffic-intelligence

comparison trafficintelligence/storage.py @ 1246:2397de73770d

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
dltrack saves after projecting coordinates
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 09 Feb 2024 17:47:33 -0500
parents 371c718e57d7
children 2aa56b101041
comparison
equal deleted inserted replaced
1245:371c718e57d7 1246:2397de73770d
1650 self.minTrackingError = config.getfloat(self.sectionHeader, 'min-tracking-error') 1650 self.minTrackingError = config.getfloat(self.sectionHeader, 'min-tracking-error')
1651 self.featureFlags = config.getboolean(self.sectionHeader, 'feature-flag') 1651 self.featureFlags = config.getboolean(self.sectionHeader, 'feature-flag')
1652 self.minFeatureEigThreshold = config.getfloat(self.sectionHeader, 'min-feature-eig-threshold') 1652 self.minFeatureEigThreshold = config.getfloat(self.sectionHeader, 'min-feature-eig-threshold')
1653 self.minFeatureTime = config.getint(self.sectionHeader, 'min-feature-time') 1653 self.minFeatureTime = config.getint(self.sectionHeader, 'min-feature-time')
1654 self.minFeatureDisplacement = config.getfloat(self.sectionHeader, 'min-feature-displacement') 1654 self.minFeatureDisplacement = config.getfloat(self.sectionHeader, 'min-feature-displacement')
1655 self.smoothingHalfWidth = config.getfloat(self.sectionHeader, 'smoothing-halfwidth')
1655 #self.updateTimer = config.getint(self.sectionHeader, 'tracker-reload-time') 1656 #self.updateTimer = config.getint(self.sectionHeader, 'tracker-reload-time')
1656 1657
1657 1658
1658 def __init__(self, filename = None): 1659 def __init__(self, filename = None):
1659 self.configFilename = filename 1660 self.configFilename = filename
1670 parentPath = Path(args.configFilename).parent 1671 parentPath = Path(args.configFilename).parent
1671 params = ProcessParameters(args.configFilename) 1672 params = ProcessParameters(args.configFilename)
1672 videoFilename = params.videoFilename 1673 videoFilename = params.videoFilename
1673 databaseFilename = params.databaseFilename 1674 databaseFilename = params.databaseFilename
1674 if params.homography is not None: 1675 if params.homography is not None:
1676 homography = params.homography
1675 invHomography = linalg.inv(params.homography) 1677 invHomography = linalg.inv(params.homography)
1676 else: 1678 else:
1679 homography = None
1677 invHomography = None 1680 invHomography = None
1678 intrinsicCameraMatrix = params.intrinsicCameraMatrix 1681 intrinsicCameraMatrix = params.intrinsicCameraMatrix
1679 distortionCoefficients = array(params.distortionCoefficients) 1682 distortionCoefficients = array(params.distortionCoefficients)
1680 undistortedImageMultiplication = params.undistortedImageMultiplication 1683 undistortedImageMultiplication = params.undistortedImageMultiplication
1681 undistort = params.undistort 1684 undistort = params.undistort
1682 firstFrameNum = params.firstFrameNum 1685 firstFrameNum = params.firstFrameNum
1683 else: 1686 else:
1684 params = None 1687 params = None
1688 homography = None
1685 invHomography = None 1689 invHomography = None
1686 undistort = False 1690 undistort = False
1687 intrinsicCameraMatrix = None 1691 intrinsicCameraMatrix = None
1688 distortionCoefficients = [] 1692 distortionCoefficients = []
1689 undistortedImageMultiplication = None 1693 undistortedImageMultiplication = None
1698 if args.databaseFilename is not None: 1702 if args.databaseFilename is not None:
1699 databaseFilename = args.databaseFilename 1703 databaseFilename = args.databaseFilename
1700 else: 1704 else:
1701 databaseFilename = params.databaseFilename 1705 databaseFilename = params.databaseFilename
1702 1706
1703 return params, videoFilename, databaseFilename, invHomography, intrinsicCameraMatrix, distortionCoefficients, undistortedImageMultiplication, undistort, firstFrameNum 1707 return params, videoFilename, databaseFilename, homography, invHomography, intrinsicCameraMatrix, distortionCoefficients, undistortedImageMultiplication, undistort, firstFrameNum
1704 1708
1705 # deprecated 1709 # deprecated
1706 class SceneParameters(object): 1710 class SceneParameters(object):
1707 def __init__(self, config, sectionName): 1711 def __init__(self, config, sectionName):
1708 from configparser import NoOptionError 1712 from configparser import NoOptionError