updates

2019-03-04 15:56:58 +01:00 · 2019-03-04 15:56:58 +01:00 · 0b3a17362c
commit 0b3a17362c
parent 5e0cce771e
1 changed files with 2 additions and 0 deletions
--- a/models.py
+++ b/models.py
@ -150,6 +150,8 @@ class YOLOLayer(nn.Module):
            p_conf = p[..., 4]  # Conf
            p_cls = p[..., 5:]  # Class

+            print(self.anchor_vec.device, self.anchor_vec, self.nA, self.nC, nG)
+
            txy, twh, mask, tcls = build_targets(targets, self.anchor_vec, self.nA, self.nC, nG)

            tcls = tcls[mask]