best possible recall fix

fwd3 · Jun 13, 2020 · 93bbf1b · 93bbf1b
1 parent 31f3310
commit 93bbf1b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/utils.py b/utils/utils.py
@@ -53,8 +53,8 @@ def check_img_size(img_size, s=32):
 
 def check_best_possible_recall(dataset, anchors, thr):
     # Check best possible recall of dataset with current anchors
-    wh = torch.tensor(np.concatenate([l[:, 3:5] * s for s, l in zip(dataset.shapes, dataset.labels)]))  # width-height
-    ratio = wh[:, None] / anchors.view(-1, 2)[None]  # ratio
+    wh = torch.tensor(np.concatenate([l[:, 3:5] * s for s, l in zip(dataset.shapes, dataset.labels)])).float()  # wh
+    ratio = wh[:, None] / anchors.view(-1, 2).cpu()[None]  # ratio
     m = torch.max(ratio, 1. / ratio).max(2)[0]  # max ratio
     bpr = (m.min(1)[0] < thr).float().mean()  # best possible recall
     mr = (m < thr).float().mean()  # match ratio