Mercurial > hg > nsaunier > traffic-intelligence

comparison python/tests/cvutils.txt @ 932:66f382852e61

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added new projection functions
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 14 Jul 2017 00:12:03 -0400
parents 7db0f2853bfd
children 8ac7f61c6e4f
comparison
equal deleted inserted replaced
931:8148991b1dab 932:66f382852e61
10 >>> undistorted = cv2.remap(img, map1, map2, interpolation=cv2.INTER_LINEAR) 10 >>> undistorted = cv2.remap(img, map1, map2, interpolation=cv2.INTER_LINEAR)
11 >>> (undistorted.shape == array([int(round(height*multiplicationFactor)), int(round(width*multiplicationFactor)), 3])).all() 11 >>> (undistorted.shape == array([int(round(height*multiplicationFactor)), int(round(width*multiplicationFactor)), 3])).all()
12 True 12 True
13 >>> imgPoints = array([[[150.,170.],[220.,340.],[340.,440.],[401.,521.]]]) 13 >>> imgPoints = array([[[150.,170.],[220.,340.],[340.,440.],[401.,521.]]])
14 >>> newCameraMatrix = cv2.getDefaultNewCameraMatrix(intrinsicCameraMatrix, (int(round(width*multiplicationFactor)), int(round(height*multiplicationFactor))), True) 14 >>> newCameraMatrix = cv2.getDefaultNewCameraMatrix(intrinsicCameraMatrix, (int(round(width*multiplicationFactor)), int(round(height*multiplicationFactor))), True)
15 >>> undistortedPoints = cv2.undistortPoints(imgPoints, intrinsicCameraMatrix, distortionCoefficients, P = newCameraMatrix).reshape(-1, 2) 15 >>> undistortedPoints = cv2.undistortPoints(imgPoints, intrinsicCameraMatrix, distortionCoefficients, P = newCameraMatrix).reshape(-1, 2) # undistort and project as if seen by new camera
16 >>> invNewCameraMatrix = linalg.inv(newCameraMatrix) 16 >>> invNewCameraMatrix = linalg.inv(newCameraMatrix)
17 >>> tmp = ones((imgPoints[0].shape[0], 3)) 17 >>> tmp = ones((imgPoints[0].shape[0], 3))
18 >>> tmp[:,:2] = undistortedPoints 18 >>> tmp[:,:2] = undistortedPoints
19 >>> reducedPoints = dot(invNewCameraMatrix, tmp.T).T 19 >>> reducedPoints = dot(invNewCameraMatrix, tmp.T).T
20 >>> origPoints = cv2.projectPoints(reducedPoints, (0.,0.,0.), (0.,0.,0.), intrinsicCameraMatrix, distortionCoefficients)[0].reshape(-1,2) 20 >>> origPoints = cv2.projectPoints(reducedPoints, (0.,0.,0.), (0.,0.,0.), intrinsicCameraMatrix, distortionCoefficients)[0].reshape(-1,2)
21 >>> (round(origPoints[1:,:]) == imgPoints[0][1:,:]).all() 21 >>> (round(origPoints[1:,:]) == imgPoints[0][1:,:]).all()
22 True 22 True
23 >>> (absolute(origPoints[0,:]-imgPoints[0][0,:])).max() < 6. 23 >>> (absolute(origPoints[0,:]-imgPoints[0][0,:])).max() < 6.
24 True 24 True
25 >>> origPoints = cvutils.projectArray(None, undistortedPoints.T, intrinsicCameraMatrix, distortionCoefficients, newCameraMatrix).T 25 >>> reducedPoints2 = cvutils.newCameraProject(undistortedPoints.T, invNewCameraMatrix)
26 >>> (reducedPoints == reducedPoints).all()
27 True
28
29 >>> undistortedPoints = cv2.undistortPoints(imgPoints, intrinsicCameraMatrix, distortionCoefficients).reshape(-1, 2) # undistort to ideal points
30 >>> origPoints = cvutils.worldToImageProject(undistortedPoints.T, intrinsicCameraMatrix, distortionCoefficients).T
26 >>> (round(origPoints[1:,:]) == imgPoints[0][1:,:]).all() 31 >>> (round(origPoints[1:,:]) == imgPoints[0][1:,:]).all()
27 True 32 True
28 >>> (absolute(origPoints[0,:]-imgPoints[0][0,:])).max() < 6. 33 >>> (absolute(origPoints[0,:]-imgPoints[0][0,:])).max() < 6.
29 True 34 True
35
36 >>> undistortedPoints = cvutils.imageToWorldProject(imgPoints[0].T, intrinsicCameraMatrix, distortionCoefficients)
37 >>> origPoints = cvutils.worldToImageProject(undistortedPoints, intrinsicCameraMatrix, distortionCoefficients).T
38 >>> (round(origPoints[1:,:]) == imgPoints[0][1:,:]).all()
39 True
40 >>> (absolute(origPoints[0,:]-imgPoints[0][0,:])).max() < 6.
41 True