From 46cd0d8cc4183806347af09f5df09231ad08b11d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 9 Nov 2020 20:59:57 +0100 Subject: [PATCH] Grid indices overflow bug fix 2 (#1551) --- utils/utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index 5841d142..08ece411 100755 --- 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