Mercurial > hg > nsaunier > traffic-intelligence

comparison trafficintelligence/moving.py @ 1245:371c718e57d7

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
interface updates
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Thu, 08 Feb 2024 16:10:54 -0500
parents 4cd8ace3552f
children 2aa56b101041
comparison
equal deleted inserted replaced
1244:00b71da2baac 1245:371c718e57d7
1398 ################## 1398 ##################
1399 1399
1400 userTypeNames = ['unknown', 1400 userTypeNames = ['unknown',
1401 'car', 1401 'car',
1402 'pedestrian', 1402 'pedestrian',
1403 'motorcycle', 1403 'motorcyclist',
1404 'cyclist', 1404 'cyclist',
1405 'bus', 1405 'bus',
1406 'truck', 1406 'truck',
1407 'automated'] 1407 'automated']
1408 1408
1845 p = self.getPositionAtInstant(t) 1845 p = self.getPositionAtInstant(t)
1846 v = self.getVelocityAtInstant(t)*velocityMultiply 1846 v = self.getVelocityAtInstant(t)*velocityMultiply
1847 arrow(p.x, p.y, v.x, v.y, width=arrowWidth) 1847 arrow(p.x, p.y, v.x, v.y, width=arrowWidth)
1848 1848
1849 def play(self, videoFilename, homography = None, undistort = False, intrinsicCameraMatrix = None, distortionCoefficients = None, undistortedImageMultiplication = 1.): 1849 def play(self, videoFilename, homography = None, undistort = False, intrinsicCameraMatrix = None, distortionCoefficients = None, undistortedImageMultiplication = 1.):
1850 cvutils.displayTrajectories(videoFilename, [self], homography = homography, firstFrameNum = self.getFirstInstant(), lastFrameNumArg = self.getLastInstant(), undistort = undistort, intrinsicCameraMatrix = intrinsicCameraMatrix, distortionCoefficients = distortionCoefficients, undistortedImageMultiplication = undistortedImageMultiplication) 1850 cvutils.displayTrajectories(videoFilename, [self], homography = homography, firstFrameNum = self.getFirstInstant(), lastFrameNum = self.getLastInstant(), undistort = undistort, intrinsicCameraMatrix = intrinsicCameraMatrix, distortionCoefficients = distortionCoefficients, undistortedImageMultiplication = undistortedImageMultiplication)
1851 1851
1852 def speedDiagnostics(self, framerate = 1., display = False, nInstantsIgnoredAtEnds=0): 1852 def speedDiagnostics(self, framerate = 1., display = False, nInstantsIgnoredAtEnds=0):
1853 speeds = framerate*self.getSpeeds(nInstantsIgnoredAtEnds) 1853 speeds = framerate*self.getSpeeds(nInstantsIgnoredAtEnds)
1854 coef = utils.linearRegression(list(range(len(speeds))), speeds) 1854 coef = utils.linearRegression(list(range(len(speeds))), speeds)
1855 print('min/5th perc speed: {} / {}\nspeed diff: {}\nspeed stdev: {}\nregression: {}'.format(min(speeds), percentile(speeds, 5), speeds[-2]-speeds[1], std(speeds), coef[0])) 1855 print('min/5th perc speed: {} / {}\nspeed diff: {}\nspeed stdev: {}\nregression: {}'.format(min(speeds), percentile(speeds, 5), speeds[-2]-speeds[1], std(speeds), coef[0]))