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 21:42:21 +02:00
parent 40221894c2
commit 03dd0b82ea
1 changed files with 15 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
train.py
View File

@ -11,19 +11,18 @@ from utils.datasets import *
from utils.utils import * from utils.utils import *
# Hyperparameters # Hyperparameters
hyp = {'k': 7.789, # loss multiple hyp = {'k': 8.4875, # loss multiple
'xy': 0.1966, # xy loss fraction 'xy': 0.079756, # xy loss fraction
'wh': 0.01144, # wh loss fraction 'wh': 0.010461, # wh loss fraction
'cls': 0.01746, # cls loss fraction 'cls': 0.02105, # cls loss fraction
'conf': 0.7745, # conf loss fraction 'conf': 0.88873, # conf loss fraction
'iou_t': 0.05732, # iou target-anchor training threshold 'iou_t': 0.10, # iou target-anchor training threshold
'lr0': 0.001467, # initial learning rate 'lr0': 0.001, # initial learning rate
'lrf': -3.904, # final learning rate = lr0 * (10 ** lrf) 'lrf': -4, # final learning rate = lr0 * (10 ** lrf)
'momentum': 0.9008, # SGD momentum 'momentum': 0.9, # SGD momentum
'weight_decay': 0.0007289, # optimizer weight decay 'weight_decay': 0.0005, # optimizer weight decay
} }
def train( def train(
cfg, cfg,
data_cfg, data_cfg,
@ -93,11 +92,11 @@ def train(
# scheduler = optim.lr_scheduler.MultiStepLR(optimizer, milestones=[218, 245], gamma=0.1, last_epoch=start_epoch - 1) # scheduler = optim.lr_scheduler.MultiStepLR(optimizer, milestones=[218, 245], gamma=0.1, last_epoch=start_epoch - 1)
# Plot lr schedule # Plot lr schedule
# y = [] y = []
# for _ in range(epochs): for _ in range(epochs):
# scheduler.step() scheduler.step()
# y.append(optimizer.param_groups[0]['lr']) y.append(optimizer.param_groups[0]['lr'])
# plt.plot(y) plt.plot(y)
# Dataset # Dataset
dataset = LoadImagesAndLabels(train_path, img_size=img_size, augment=True) dataset = LoadImagesAndLabels(train_path, img_size=img_size, augment=True)