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: