From 1bc4f89bca451cacbc55756103adb763222c33e1 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 26 Aug 2018 19:34:04 +0200 Subject: [PATCH] updates --- utils/datasets.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index 277b7001..ed71c31b 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -150,7 +150,8 @@ class ListDataset(): # for training labels = np.array([]) # Augment image and labels - img, labels, M = random_affine(img, targets=labels, degrees=(-5, 5), translate=(0.2, 0.2), scale=(0.8, 1.2)) # RGB + img, labels, M = random_affine(img, targets=labels, degrees=(-5, 5), translate=(0.2, 0.2), + scale=(0.8, 1.2)) # RGB plotFlag = False if plotFlag: