Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2018-12-31 12:33:34 +01:00
parent 36a06a1e90
commit 17a02ae3e4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -168,7 +168,7 @@ class YOLOLayer(nn.Module):
p_boxes = None
if batch_report:
# Predictd boxes: add offset and scale with anchors (in grid space, i.e. 0-13)
# Predicted boxes: add offset and scale with anchors (in grid space, i.e. 0-13)
gx = x.data + self.grid_x[:, :, :nG, :nG]
gy = y.data + self.grid_y[:, :, :nG, :nG]
p_boxes = torch.stack((gx - width / 2,