Mercurial Hosting > traffic-intelligence
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()