From b079c1b10c935aa00e37a912fc16a3ca2039c599 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 15 Dec 2018 21:01:14 +0100 Subject: [PATCH] updates --- utils/datasets.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/datasets.py b/utils/datasets.py index 940c2adb..cfcb19b1 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -116,7 +116,7 @@ class load_images_and_labels(): # for training augment_hsv = True if self.augment and augment_hsv: # SV augmentation by 50% - fraction = 0.25 + fraction = 0.50 img_hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV) S = img_hsv[:, :, 1].astype(np.float32) V = img_hsv[:, :, 2].astype(np.float32) @@ -153,7 +153,7 @@ class load_images_and_labels(): # for training # Augment image and labels if self.augment: - img, labels, M = random_affine(img, labels, degrees=(-5, 5), translate=(0.05, 0.05), scale=(0.95, 1.05)) + img, labels, M = random_affine(img, labels, degrees=(-5, 5), translate=(0.10, 0.10), scale=(0.90, 1.10)) plotFlag = False if plotFlag: