From 693c06b26ca24f3cb24241ed3cf2a3cf03899341 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 18 Apr 2020 12:07:44 -0700 Subject: [PATCH] bug fix issues/1067 --- utils/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index e2af1aae..bb2b21ae 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -601,7 +601,7 @@ def letterbox(img, new_shape=(416, 416), color=(114, 114, 114), auto=True, scale new_shape = (new_shape, new_shape) # Scale ratio (new / old) - r = max(new_shape) / max(shape) + r = min(new_shape[0] / shape[0], new_shape[1] / shape[1]) if not scaleup: # only scale down, do not scale up (for better test mAP) r = min(r, 1.0)