Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates 

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2019-09-03 17:23:59 +02:00
parent 447292eb36
commit 976eea04bd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -90,7 +90,7 @@ def create_modules(module_defs, img_size, arc):
elif arc == 'uFBCE': # unified FocalBCE (5120 obj, 80 classes) elif arc == 'uFBCE': # unified FocalBCE (5120 obj, 80 classes)
b = [0, -6.5] b = [0, -6.5]
elif arc == 'uFCE': # unified FocalCE (64 cls, 1 background + 80 classes) 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 = module_list[-1][0].bias.view(len(mask), -1) # 255 to 3x85
bias[:, 4] += b[0] - bias[:, 4].mean() # obj bias[:, 4] += b[0] - bias[:, 4].mean() # obj