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