Grid indices overflow bug fix 2 (#1551)

46cd0d8c · Glenn Jocher · GitHub · 95460570 · 46cd0d8c
隐藏空白更改
内联并排

Showing with 0 addition and 1 deletion

utils/utils.py utils/utils.py +0 -1

未找到文件。
--- 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