From c4f9e3891e6d87f4bd0d51d54dbe479231e89717 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 25 Aug 2019 03:03:35 +0200 Subject: [PATCH] updates --- models.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/models.py b/models.py index fb87a649..bcfedb3c 100755 --- a/models.py +++ b/models.py @@ -78,11 +78,11 @@ def create_modules(module_defs, img_size, arc): # Initialize preceding Conv2d() bias (https://arxiv.org/pdf/1708.02002.pdf section 3.3) try: if arc == 'default': - b = [-5, -4] # obj, cls + b = [-4, -3.6] # obj, cls elif arc == 'uCE': # unified CE (1 background + 80 classes) - b = [7, 0] # obj, cls + b = [10, -0.1] # obj, cls elif arc == 'uBCE': # unified BCE (80 classes) - b = [0, -3.5] # obj, cls + b = [0, -8.5] # obj, cls bias = module_list[-1][0].bias.view(len(mask), -1) # 255 to 3x85 bias[:, 4] += b[0] # obj