This commit is contained in:
Glenn Jocher 2018-08-26 19:34:04 +02:00
parent 8a1d1b76c0
commit 1bc4f89bca
1 changed files with 2 additions and 1 deletions

View File

@ -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: