From 17a02ae3e464277ff1aefe26947d62b863ed34bf Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 31 Dec 2018 12:33:34 +0100 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 6ca97048..287f4904 100755 --- a/models.py +++ b/models.py @@ -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,