From e73e24744265e8fb9ce7982ff61fbde30c23d911 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 15 Jul 2019 16:07:25 +0200
Subject: [PATCH] updates

---
 utils/utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/utils/utils.py b/utils/utils.py
index d02c93ec..7fda5289 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -130,7 +130,8 @@ def scale_coords(img1_shape, coords, img0_shape):
     coords[:, [0, 2]] -= (img1_shape[1] - img0_shape[1] * gain) / 2  # x padding
     coords[:, [1, 3]] -= (img1_shape[0] - img0_shape[0] * gain) / 2  # y padding
     coords[:, :4] /= gain
-    coords[:, :4] = coords[:, :4].clamp(min=0)
+    coords[:, [0, 2]] = coords[:, [0, 2]].clamp(min=0, max=img0_shape[1])  # clip x
+    coords[:, [1, 3]] = coords[:, [1, 3]].clamp(min=0, max=img0_shape[0])  # clip y
     return coords