From 5ca987eaed05f6bef18cc658333dbb0c3e4800aa Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 4 Mar 2019 16:01:23 +0100 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 8907b626..2e45ddb4 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -243,7 +243,7 @@ def build_targets(target, anchor_vec, nA, nC, nG): tcls = torch.ByteTensor(nB, nA, nG, nG, nC).fill_(0) # nC = number of classes for b in range(nB): - t = target[b] + t = target[b].cpu() nTb = len(t) # number of targets if nTb == 0: continue @@ -257,7 +257,7 @@ def build_targets(target, anchor_vec, nA, nC, nG): box1 = gwh box2 = anchor_vec.unsqueeze(1) - print(box1.device,box2.device) + print(box1.device, box2.device) inter_area = torch.min(box1, box2).prod(2) iou = inter_area / (box1.prod(1) + box2.prod(2) - inter_area + 1e-16)