Feature: More augmentation

Flippchen · Feb 8, 2024 · 0720f69 · 0720f69
1 parent b223133
commit 0720f69
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utilities/tools.py b/utilities/tools.py
@@ -161,6 +161,7 @@ def create_augmentation_layer(img_height: int, img_width: int) -> keras.Sequenti
             layers.RandomRotation(0.2),
             layers.RandomZoom(0.1),
             layers.RandomContrast(0.1),
+            layers.GaussianNoise(0.1)
         ]
     )