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

updates

This commit is contained in:
Glenn Jocher 2020-03-04 14:55:56 -08:00
parent 1d45ec84bc
commit 4a5159710f
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

@ -89,7 +89,7 @@ def create_modules(module_defs, img_size, arc):
try: try:
p = math.log(1 / (modules.nc - 0.99)) # class probability -> sigmoid(p) = 1/nc p = math.log(1 / (modules.nc - 0.99)) # class probability -> sigmoid(p) = 1/nc
if arc == 'default' or arc == 'Fdefault': # default if arc == 'default' or arc == 'Fdefault': # default
b = [-5.0, p] # obj, cls b = [-4.5, p] # obj, cls
elif arc == 'uBCE': # unified BCE (80 classes) elif arc == 'uBCE': # unified BCE (80 classes)
b = [0, -9.0] b = [0, -9.0]
elif arc == 'uCE': # unified CE (1 background + 80 classes) elif arc == 'uCE': # unified CE (1 background + 80 classes)