From b636f7f7ab66bd40d8f86f60e2fe31d8c0d54ea6 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 30 Dec 2019 11:57:36 -0800
Subject: [PATCH] updates

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

diff --git a/utils/utils.py b/utils/utils.py
index feba4183..06d0b799 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -760,7 +760,7 @@ def kmeans_targets(path='../coco/trainvalno5k.txt', n=9, img_size=416):  # from
     # ax[0].plot(np.arange(1, 21), np.array(d) ** 2, marker='.')
 
     # Measure IoUs
-    iou = torch.stack([wh_iou(torch.Tensor(wh).T, torch.Tensor(x).T) for x in k], 0)
+    iou = wh_iou(torch.Tensor(wh), torch.Tensor(k))
     biou = iou.max(0)[0]  # closest anchor IoU
     print('Best possible recall: %.3f' % (biou > 0.2635).float().mean())  # BPR (best possible recall)