Grid indices overflow bug fix (#1551)

2020-11-06 13:38:13 +01:00 · 2020-11-06 13:38:13 +01:00 · ac601cf681
commit ac601cf681
parent cf652962fd
1 changed files with 1 additions and 0 deletions
--- a/utils/utils.py
+++ b/utils/utils.py
@ -465,6 +465,7 @@ def build_targets(p, targets, model):

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