This commit is contained in:
Glenn Jocher 2019-07-10 22:11:48 +02:00
parent 4f6ef59d92
commit 3373006d0e
1 changed files with 1 additions and 1 deletions

View File

@ -15,7 +15,7 @@ from utils.utils import *
hyp = {'giou': 1.666, # giou loss gain hyp = {'giou': 1.666, # giou loss gain
'xy': 4.062, # xy loss gain 'xy': 4.062, # xy loss gain
'wh': 0.1845, # wh loss gain 'wh': 0.1845, # wh loss gain
'cls': 42.6, # cls loss gain 'cls': 1.0, # cls loss gain
'cls_pw': 3.34, # cls BCELoss positive_weight 'cls_pw': 3.34, # cls BCELoss positive_weight
'obj': 12.61, # obj loss gain 'obj': 12.61, # obj loss gain
'obj_pw': 8.338, # obj BCELoss positive_weight 'obj_pw': 8.338, # obj BCELoss positive_weight