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

updates

This commit is contained in:
Glenn Jocher 2019-06-12 14:35:10 +02:00
parent 81b4a7833f
commit c5cb3c8a9e
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -12,7 +12,7 @@ from utils.datasets import *
from utils.utils import * from utils.utils import *
# Hyperparameters: train.py --evolve --epochs 2 --img-size 320, Metrics: 0.204 0.302 0.175 0.234 (square smart) # Hyperparameters: train.py --evolve --epochs 2 --img-size 320, Metrics: 0.204 0.302 0.175 0.234 (square smart)
hyp = {'xy': 0.2, # xy loss gain hyp = {'xy': 0.2, # xy loss gain (giou is about 0.02)
'wh': 0.1, # wh loss gain 'wh': 0.1, # wh loss gain
'cls': 0.04, # cls loss gain 'cls': 0.04, # cls loss gain
'conf': 4.5, # conf loss gain 'conf': 4.5, # conf loss gain

4
utils/datasets.py
View File

@ -230,7 +230,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
augment_hsv = True augment_hsv = True
if self.augment and augment_hsv: if self.augment and augment_hsv:
# SV augmentation by 50% # SV augmentation by 50%
fraction = 0.25 # must be < 1.0 fraction = 0.50 # must be < 1.0
img_hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV) # hue, sat, val img_hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV) # hue, sat, val
S = img_hsv[:, :, 1].astype(np.float32) # saturation S = img_hsv[:, :, 1].astype(np.float32) # saturation
V = img_hsv[:, :, 2].astype(np.float32) # value V = img_hsv[:, :, 2].astype(np.float32) # value
@ -269,7 +269,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
# Augment image and labels # Augment image and labels
if self.augment: if self.augment:
img, labels = random_affine(img, labels, degrees=(-5, 5), translate=(0.0, 0.0), scale=(1.0, 1.0)) img, labels = random_affine(img, labels, degrees=(-5, 5), translate=(0.10, 0.10), scale=(0.90, 1.10))
nL = len(labels) # number of labels nL = len(labels) # number of labels
if nL: if nL: