diff trafficintelligence/cvutils.py @ 1029:c6cf75a2ed08

reorganization of imports
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Mon, 18 Jun 2018 22:50:59 -0400
parents cc5cb04b04b0
children 956a66096e91
line wrap: on
line diff
--- a/trafficintelligence/cvutils.py	Fri Jun 15 11:19:10 2018 -0400
+++ b/trafficintelligence/cvutils.py	Mon Jun 18 22:50:59 2018 -0400
@@ -1,7 +1,16 @@
 #! /usr/bin/env python
 '''Image/Video utilities'''
 
-from trafficintelligence import utils, moving
+from sys import stdout
+from os import listdir
+from subprocess import run
+from math import floor, log10, ceil
+from time import time
+
+from numpy import dot, array, append, float32, loadtxt, savetxt, append, zeros, ones, identity, abs as npabs, logical_and, unravel_index, sum as npsum, isnan, mgrid, median, floor as npfloor, ceil as npceil
+from numpy.linalg import inv
+from matplotlib.mlab import find
+from matplotlib.pyplot import imread, imsave, imshow, figure, subplot
 
 try:
     import cv2
@@ -16,15 +25,7 @@
     print('Scikit-image library could not be loaded (HoG-based classification methods will not be available)')
     skimageAvailable = False
     
-from sys import stdout
-from os import listdir
-from subprocess import run
-from math import floor, log10, ceil
-
-from numpy import dot, array, append, float32, loadtxt, savetxt, append, zeros, ones, identity, abs as npabs, logical_and, unravel_index, sum as npsum, isnan, mgrid, median, floor as npfloor, ceil as npceil
-from numpy.linalg import inv
-from matplotlib.mlab import find
-from matplotlib.pyplot import imread, imsave
+from trafficintelligence import utils, moving
 
 videoFilenameExtensions = ['mov', 'avi', 'mp4', 'MOV', 'AVI', 'MP4']
 trackerExe = 'feature-based-tracking'
@@ -103,10 +104,9 @@
 
     def cvImshow(windowName, img, rescale = 1.0):
         'Rescales the image (in particular if too large)'
-        from cv2 import resize
         if rescale != 1.:
             size = (int(round(img.shape[1]*rescale)), int(round(img.shape[0]*rescale)))
-            resizedImg = resize(img, size)
+            resizedImg = cv2.resize(img, size)
             cv2.imshow(windowName, resizedImg)
         else:
             cv2.imshow(windowName, img)
@@ -303,7 +303,6 @@
         if undistort:
             cmd += ['--undistort', 'true']
             if intrinsicCameraMatrix is not None: # we currently have to save a file
-                from time import time
                 intrinsicCameraFilename = '/tmp/intrinsic-{}.txt'.format(time())
                 savetxt(intrinsicCameraFilename, intrinsicCameraMatrix)
                 cmd += ['--intrinsic-camera-filename', intrinsicCameraFilename]
@@ -628,7 +627,6 @@
         inputImg = transform.resize(bwImg, rescaleSize)
         features = hog(inputImg, orientations, pixelsPerCell, cellsPerBlock, blockNorm, visualize, transformSqrt, True)
         if visualize:
-            from matplotlib.pyplot import imshow, figure, subplot
             hogViz = features[1]
             features = features[0]
             figure()