repo/traffic-intelligence: python/tests/cvutils.txt annotate

annotate python/tests/cvutils.txt @ 929:be28a3538dc9

work in progress on projection

author	Nicolas Saunier <nicolas.saunier@polymtl.ca>
date	Wed, 12 Jul 2017 18:00:53 -0400
parents	52aa03260f03
children	7db0f2853bfd

rev	line source
807 52aa03260f03 reversed all code to OpenCV 2.4.13 Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: diff changeset	1 >>> import cv2, cvutils
929 be28a3538dc9 work in progress on projection Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: 807 diff changeset	2 >>> from numpy import array, round, ones
807 52aa03260f03 reversed all code to OpenCV 2.4.13 Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: diff changeset	3 >>> img = cv2.imread("../samples/val-dor-117-111.png")
52aa03260f03 reversed all code to OpenCV 2.4.13 Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: diff changeset	4 >>> width = img.shape[1]
52aa03260f03 reversed all code to OpenCV 2.4.13 Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: diff changeset	5 >>> height = img.shape[0]
52aa03260f03 reversed all code to OpenCV 2.4.13 Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: diff changeset	6 >>> intrinsicCameraMatrix = array([[ 377.42, 0. , 639.12], [ 0. , 378.43, 490.2 ], [ 0. , 0. , 1. ]])
929 be28a3538dc9 work in progress on projection Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: 807 diff changeset	7 >>> distortionCoefficients = array([-0.11759321, 0.0148536, 0.00030756, -0.00020578, -0.00091816])
be28a3538dc9 work in progress on projection Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: 807 diff changeset	8 >>> distortionCoefficients = array([-0.11759321, 0., 0., 0., 0.])
be28a3538dc9 work in progress on projection Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: 807 diff changeset	9 >>> multiplicationFactor = 1.31
be28a3538dc9 work in progress on projection Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: 807 diff changeset	10 >>> [map1, map2] = cvutils.computeUndistortMaps(width, height, multiplicationFactor, intrinsicCameraMatrix, distortionCoefficients)
807 52aa03260f03 reversed all code to OpenCV 2.4.13 Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: diff changeset	11 >>> undistorted = cv2.remap(img, map1, map2, interpolation=cv2.INTER_LINEAR)
929 be28a3538dc9 work in progress on projection Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: 807 diff changeset	12 >>> (undistorted.shape == array([int(round(heightmultiplicationFactor)), int(round(widthmultiplicationFactor)), 3])).all()
807 52aa03260f03 reversed all code to OpenCV 2.4.13 Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: diff changeset	13 True
929 be28a3538dc9 work in progress on projection Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: 807 diff changeset	14 >>> imgPoints = array([[[150.,170.],[220.,340.],[340.,440.],[401.,521.]]])
be28a3538dc9 work in progress on projection Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: 807 diff changeset	15 >>> newCameraMatrix = cv2.getDefaultNewCameraMatrix(intrinsicCameraMatrix, (int(round(widthmultiplicationFactor)), int(round(heightmultiplicationFactor))), True)
be28a3538dc9 work in progress on projection Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: 807 diff changeset	16 >>> undistortedPoints = cv2.undistortPoints(imgPoints, intrinsicCameraMatrix, distortionCoefficients, P = newCameraMatrix).reshape(-1, 2)
be28a3538dc9 work in progress on projection Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: 807 diff changeset	17 >>> tmp = ones((imgPoints[0].shape[0], 3))
be28a3538dc9 work in progress on projection Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: 807 diff changeset	18 >>> tmp[:,:2] = undistortedPoints
be28a3538dc9 work in progress on projection Nicolas Saunier <nicolas.saunier@polymtl.ca> parents: 807 diff changeset	19 >>> origPoints = cv2.projectPoints(tmp, (0.,0.,0.), (0.,0.,0.), intrinsicCameraMatrix, distortionCoefficients)[0].reshape(-1,2)