fix bug

2022-07-05 19:44:14 +08:00 · 2022-07-05 19:44:14 +08:00 · fdf9e05803
commit fdf9e05803
parent b1dfad4a21
1 changed files with 1 additions and 1 deletions
--- 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)