add GaussNoise to augmentation step

2024-04-28 16:09:37 +01:00
parent 15b80da3c3
commit 9364d97bd7
1 changed files with 1 additions and 0 deletions
--- 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]),