From 976eea04bd3b602acaf09d9faa075c9a390c86eb Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 3 Sep 2019 17:23:59 +0200 Subject: [PATCH] updates Signed-off-by: Glenn Jocher --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 6982231c..abca7e61 100755 --- a/models.py +++ b/models.py @@ -90,7 +90,7 @@ def create_modules(module_defs, img_size, arc): elif arc == 'uFBCE': # unified FocalBCE (5120 obj, 80 classes) b = [0, -6.5] elif arc == 'uFCE': # unified FocalCE (64 cls, 1 background + 80 classes) - b = [7, -0.1] + b = [7.7, -1.1] bias = module_list[-1][0].bias.view(len(mask), -1) # 255 to 3x85 bias[:, 4] += b[0] - bias[:, 4].mean() # obj