From 9364d97bd79914095abaee0cd8e36127fa266d6d Mon Sep 17 00:00:00 2001
From: Kenneth <kennethnym@outlook.com>
Date: Sun, 28 Apr 2024 16:09:37 +0100
Subject: [PATCH] add GaussNoise to augmentation step

---
 augmentation.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/augmentation.py b/augmentation.py
index ea6adcb..11d7112 100644
--- a/augmentation.py
+++ b/augmentation.py
@@ -8,6 +8,7 @@ preprocess_training = a.Compose(
     [
         a.augmentations.PadIfNeeded(min_width=CROP_SIZE, min_height=CROP_SIZE),
         a.RandomCrop(width=CROP_SIZE, height=CROP_SIZE),
+        a.GaussNoise(),
         a.Flip(p=0.5),
         a.RandomRotate90(p=0.5),
         a.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),