Add 90 rotation to possible data augmentation

ndrplz · Oct 15, 2017 · da81c07 · da81c07
1 parent d270db7
commit da81c07
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/capstone_traffic_light_classifier/traffic_light_dataset.py b/capstone_traffic_light_classifier/traffic_light_dataset.py
@@ -91,9 +91,13 @@ def coin_flip_is_head():
 
         for b in range(batch.shape[0]):
             if coin_flip_is_head():
-                batch[b] = np.fliplr(batch[b])
+                batch[b] = np.fliplr(batch[b])    # flip horizontally (mirroring)
             if coin_flip_is_head():
-                batch[b] = np.flipud(batch[b])
+                batch[b] = np.flipud(batch[b])    # flip vertically
+            if coin_flip_is_head():
+                batch[b] = np.rot90(batch[b], 3)  # rotate 90 degree clockwise
+            if coin_flip_is_head():
+                batch[b] = np.rot90(batch[b], 1)  # rotate 90 degree counter-clockwise
 
         return batch