updates

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2019-09-03 17:23:59 +02:00 · 2019-09-03 17:23:59 +02:00 · 976eea04bd
parent 447292eb36
commit 976eea04bd
1 changed files with 1 additions and 1 deletions
--- 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