apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Grid indices overflow bug fix (#1551)

Browse Source
This commit is contained in:
Glenn Jocher 2020-11-06 13:38:13 +01:00 committed by GitHub
parent cf652962fd
commit ac601cf681
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
utils/utils.py
View File

@ -465,6 +465,7 @@ def build_targets(p, targets, model):
# Append # Append
indices.append((b, a, gj, gi)) # image, anchor, grid indices 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 tbox.append(torch.cat((gxy - gij, gwh), 1)) # box
anch.append(anchors[a]) # anchors anch.append(anchors[a]) # anchors
tcls.append(c) # class tcls.append(c) # class