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

updates

This commit is contained in:
Glenn Jocher 2018-08-26 11:30:46 +02:00
parent 67ee4f0c0d
commit a27276f055
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils/datasets.py
View File

@ -105,7 +105,7 @@ class ListDataset(): # for training
if img is None:
continue
augment_hsv = False
augment_hsv = True
if augment_hsv:
# SV augmentation by 50%
fraction = 0.50
@ -144,7 +144,7 @@ class ListDataset(): # for training
labels = np.array([])
# Augment image and labels
# img, labels, M = random_affine(img, targets=labels, degrees=(-5, 5), translate=(0.1, 0.1), scale=(0.8, 1.2)) # RGB
img, labels, M = random_affine(img, targets=labels, degrees=(-10, 10), translate=(0.2, 0.2), scale=(0.8, 1.2)) # RGB
plotFlag = False
if plotFlag:
@ -158,7 +158,7 @@ class ListDataset(): # for training
labels[:, 1:5] = xyxy2xywh(labels[:, 1:5].copy()) / height
# random left-right flip
lr_flip = False
lr_flip = True
if lr_flip & (random.random() > 0.5):
img = np.fliplr(img)
if nL > 0: