From 7b6bd39c9e8170d614fe8d20c1ed3e6a89d170c0 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 30 Dec 2019 13:46:21 -0800
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 3464a2fe..1a9fe5db 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -762,7 +762,8 @@ def kmean_anchors(path='data/coco64.txt', n=9, img_size=(288, 640)):  # from uti
 
     # Measure IoUs
     iou = wh_iou(torch.Tensor(wh), torch.Tensor(k))
-    biou = iou.max(0)[0]  # closest anchor IoU
+    biou = iou.max(1)[0]  # closest anchor IoU
+    print(biou.shape)
     print('Best Possible Recall (BPR): %.3f' % (biou > 0.225).float().mean())  # BPR (best possible recall)
 
     # Print