From fdf9e05803f5e2effabda2125e4ae998334cd2a1 Mon Sep 17 00:00:00 2001 From: duoduo Date: Tue, 5 Jul 2022 19:44:14 +0800 Subject: [PATCH] fix bug --- module/loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/module/loss.py b/module/loss.py index 9658eb9..446411a 100644 --- a/module/loss.py +++ b/module/loss.py @@ -111,7 +111,7 @@ class DetectorLoss(nn.Module): tobj = torch.zeros_like(pobj) factor = torch.ones_like(pobj) * 0.75 - if len(gt_box[0]) > 0: + if len(gt_box) > 0: # 计算检测框回归loss b, gx, gy = ps_index[0] ptbox = torch.ones((preg[b, gy, gx].shape)).to(self.device)