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 19:25:41 +02:00
parent 97c488f8ef
commit 02d6b2f9c5
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 *
# Hyperparameters
hyp = {'k': 10.11, # loss multiple
'xy': 0.1509, # xy loss fraction
'wh': 0.008077, # wh loss fraction
'cls': 0.01082, # cls loss fraction
'conf': 0.8302, # conf loss fraction
'iou_t': 0.05892, # iou target-anchor training threshold
'lr0': 0.001475, # initial learning rate
'lrf': -3.371, # final learning rate = lr0 * (10 ** lrf)
'momentum': 0.8733, # SGD momentum
'weight_decay': 0.0006636, # optimizer weight decay
hyp = {'k': 7.789, # loss multiple
'xy': 0.1966, # xy loss fraction
'wh': 0.01144, # wh loss fraction
'cls': 0.01746, # cls loss fraction
'conf': 0.7745, # conf loss fraction
'iou_t': 0.05732, # iou target-anchor training threshold
'lr0': 0.001467, # initial learning rate
'lrf': -3.904, # final learning rate = lr0 * (10 ** lrf)
'momentum': 0.9008, # SGD momentum
'weight_decay': 0.0007289, # optimizer weight decay
}