diff scripts/classify-objects.py @ 904:8f60ecfc2f06

work in progress, almost ready
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Thu, 22 Jun 2017 18:08:46 -0400
parents c69a8defe5c3
children 0e017178f7ab
line wrap: on
line diff
--- a/scripts/classify-objects.py	Thu Jun 22 17:02:03 2017 -0400
+++ b/scripts/classify-objects.py	Thu Jun 22 18:08:46 2017 -0400
@@ -71,10 +71,11 @@
 width = int(capture.get(cv2.cv.CV_CAP_PROP_FRAME_WIDTH))
 height = int(capture.get(cv2.cv.CV_CAP_PROP_FRAME_HEIGHT))
 
+if undistort: # setup undistortion
+    [map1, map2] = cvutils.computeUndistortMaps(width, height, undistortedImageMultiplication, intrinsicCameraMatrix, distortionCoefficients)
+
 pastObjects = []
 currentObjects = []
-if undistort: # setup undistortion
-    [map1, map2] = cvutils.computeUndistortMaps(width, height, undistortedImageMultiplication, intrinsicCameraMatrix, distortionCoefficients)
 if capture.isOpened():
     ret = True
     frameNum = timeInterval.first