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

updates

This commit is contained in:
Glenn Jocher 2019-04-18 16:17:44 +02:00
parent b7c5eb1503
commit c68093ee7d
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
train.py
View File

@ -11,16 +11,16 @@ from utils.datasets import *
from utils.utils import * from utils.utils import *
# Hyperparameters # Hyperparameters
hyp = {'k': 6.927, # loss multiple hyp = {'k': 7.104, # loss multiple
'xy': 0.07556, # xy loss fraction 'xy': 0.0928, # xy loss fraction
'wh': 0.008074, # wh loss fraction 'wh': 0.007779, # wh loss fraction
'cls': 0.01113, # cls loss fraction 'cls': 0.01283, # cls loss fraction
'conf': 0.9052, # conf loss fraction 'conf': 0.8866, # conf loss fraction
'iou_t': 0.06154, # iou target-anchor training threshold 'iou_t': 0.06478, # iou target-anchor training threshold
'lr0': 0.001136, # initial learning rate 'lr0': 0.001132, # initial learning rate
'lrf': -2.52, # final learning rate = lr0 * (10 ** lrf) 'lrf': -2.607, # final learning rate = lr0 * (10 ** lrf)
'momentum': 0.9, # SGD momentum 'momentum': 0.9016, # SGD momentum
'weight_decay': 0.0005, # optimizer weight decay 'weight_decay': 0.000531, # optimizer weight decay
} }