From 44ea6984f9ba920d79243e04304993fe497ea938 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 19 Aug 2019 18:03:33 +0200 Subject: [PATCH] updates --- models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models.py b/models.py index 9cd7d9ff..000f5570 100755 --- a/models.py +++ b/models.py @@ -77,8 +77,8 @@ def create_modules(module_defs, img_size): # Initialize preceding Conv2d() bias (https://arxiv.org/pdf/1708.02002.pdf section 3.3) bias = module_list[-1][0].bias.view(len(mask), -1) # 255 to 3x85 - bias[:, 4] -= 3.0 # obj - bias[:, 5:] -= 0.3 # cls + bias[:, 4] -= 5.0 # obj + bias[:, 5:] -= 4.0 # cls module_list[-1][0].bias = torch.nn.Parameter(bias.view(-1)) # for l in model.yolo_layers: # print pretrained biases