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

updates

This commit is contained in:
Glenn Jocher 2019-12-17 12:26:42 -08:00
parent a5677d3f90
commit adc2663fe7
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

@ -86,7 +86,7 @@ def create_modules(module_defs, img_size, arc):
elif arc == 'default': # default no pw (40 cls, 80 obj) elif arc == 'default': # default no pw (40 cls, 80 obj)
b = [-5.5, -5.0] b = [-5.5, -5.0]
elif arc == 'uBCE': # unified BCE (80 classes) elif arc == 'uBCE': # unified BCE (80 classes)
b = [0, -8.5] b = [0, -9.0]
elif arc == 'uCE': # unified CE (1 background + 80 classes) elif arc == 'uCE': # unified CE (1 background + 80 classes)
b = [10, -0.1] b = [10, -0.1]
elif arc == 'Fdefault': # Focal default no pw (28 cls, 21 obj, no pw) elif arc == 'Fdefault': # Focal default no pw (28 cls, 21 obj, no pw)