Mercurial Hosting > traffic-intelligence
comparison trafficintelligence/storage.py @ 1246:2397de73770d
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 |