updates

2020-01-12 12:01:58 -08:00 · 2020-01-12 12:01:58 -08:00 · b890ccecfc
commit b890ccecfc
parent aeac9b78eb
1 changed files with 1 additions and 1 deletions
--- a/models.py
+++ b/models.py
@ -206,7 +206,7 @@ class YOLOLayer(nn.Module):
            if self.nc == 1:
                io[..., 5] = 1  # single-class model https://github.com/ultralytics/yolov3/issues/235

-            # reshape from [1, 3, 13, 13, 85] to [1, 507, 84], remove obj_conf
+            # reshape from [1, 3, 13, 13, 85] to [1, 507, 85]
            return io.view(bs, -1, self.no), p