Grid indices overflow bug fix 2 (#1551)

2020-11-09 20:59:57 +01:00 · 2020-11-09 20:59:57 +01:00 · 46cd0d8cc4
commit 46cd0d8cc4
parent 95460570d9
1 changed files with 0 additions and 1 deletions
--- a/utils/utils.py
+++ b/utils/utils.py
@ -464,7 +464,6 @@ def build_targets(p, targets, model):
        gi, gj = gij.T  # grid xy indices

        # Append
-        indices.append((b, a, gj, gi))  # image, anchor, grid indices
        indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1)))  # image, anchor, grid indices
        tbox.append(torch.cat((gxy - gij, gwh), 1))  # box
        anch.append(anchors[a])  # anchors