Glenn Jocher
GitHub
2 years ago
1 changed files with 3 additions and 3 deletions
+ 3
- 3
utils/loss.py
View File
| device=self.device).float() * g # offsets | device=self.device).float() * g # offsets | ||||
| for i in range(self.nl): | for i in range(self.nl): | ||||
| anchors = self.anchors[i] | |||||
| gain[2:6] = torch.tensor(p[i].shape)[[3, 2, 3, 2]] # xyxy gain | |||||
| anchors, shape = self.anchors[i], p[i].shape | |||||
| gain[2:6] = torch.tensor(shape)[[3, 2, 3, 2]] # xyxy gain | |||||
| # Match targets to anchors | # Match targets to anchors | ||||
| t = targets * gain # shape(3,n,7) | t = targets * gain # shape(3,n,7) | ||||
| gi, gj = gij.T # grid indices | gi, gj = gij.T # grid indices | ||||
| # Append | # Append | ||||
| indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1))) # image, anchor, grid indices | |||||
| indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1))) # image, anchor, grid | |||||
| 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 |
Loading…