diff trafficintelligence/cvutils.py @ 1125:b358bed29ab4

updates and bugs
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Thu, 30 Jan 2020 10:17:07 -0500
parents 956a66096e91
children 7eb972942f22
line wrap: on
line diff
--- a/trafficintelligence/cvutils.py	Tue Jun 25 17:23:43 2019 -0400
+++ b/trafficintelligence/cvutils.py	Thu Jan 30 10:17:07 2020 -0500
@@ -266,7 +266,7 @@
         ymax = max(y)
         xMm = px * (xmax - xmin)
         yMm = py * (ymax - ymin)
-        a = max(ymax - ymin + (2 * yMm), xmax - (xmin + 2 * xMm))
+        a = max(ymax - ymin + (2 * yMm), xmax - xmin + (2 * xMm))
         yCropMin = int(max(0, .5 * (ymin + ymax - a)))
         yCropMax = int(min(height - 1, .5 * (ymin + ymax + a)))
         xCropMin = int(max(0, .5 * (xmin + xmax - a)))
@@ -276,7 +276,7 @@
     def imageBox(img, obj, frameNum, width, height, px = 0.2, py = 0.2, minNPixels = 800):
         'Computes the bounding box of object at frameNum'
         yCropMin, yCropMax, xCropMin, xCropMax = imageBoxSize(obj, frameNum, width, height, px, py)
-        if yCropMax != yCropMin and xCropMax != xCropMin and (yCropMax - yCropMin) * (xCropMax - xCropMin) > minNPixels:
+        if yCropMax > yCropMin and xCropMax > xCropMin and (yCropMax - yCropMin) * (xCropMax - xCropMin) > minNPixels:
             return img[yCropMin : yCropMax, xCropMin : xCropMax]
         else:
             return None