From dd2d713484c2b907604f906595fa7f72e4d8c82e Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 3 May 2019 20:51:30 +0200
Subject: [PATCH] updates

---
 models.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/models.py b/models.py
index 33176bb9..7bf737a3 100755
--- a/models.py
+++ b/models.py
@@ -165,8 +165,8 @@ class YOLOLayer(nn.Module):
             io[..., 4:] = torch.sigmoid(io[..., 4:])  # p_conf, p_cls
             # io[..., 5:] = F.softmax(io[..., 5:], dim=4)  # p_cls
             io[..., :4] *= self.stride
-            if self.nc == 1:  # single-class model https://github.com/ultralytics/yolov3/issues/235
-                io[..., 5] = 1
+            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, 85]
             return io.view(bs, -1, 5 + self.nc), p