From 4f78eec83ed132839de5d49d8e311725ca2cf272 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 15 Aug 2019 16:11:34 +0200
Subject: [PATCH] updates

---
 utils/utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/utils/utils.py b/utils/utils.py
index 867def60..9dd5cf4e 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -583,9 +583,9 @@ def kmeans_targets(path='data/coco_64img.txt', n=9, img_size=416):  # from utils
     # Get label wh
     dataset = LoadImagesAndLabels(path, augment=True, rect=True)
     for s, l in zip(dataset.shapes, dataset.labels):
-        l[:, [1, 3]] *= s[0]
+        l[:, [1, 3]] *= s[0]  # normalized to pixels
         l[:, [2, 4]] *= s[1]
-        l[:, 1:] *= img_size / max(s)  # nominal img_size for training here
+        l[:, 1:] *= img_size / max(s)  # nominal img_size for training
     wh = np.concatenate(dataset.labels, 0)[:, 3:5]  # wh from cxywh
 
     # Kmeans calculation