Merge pull request #35 from stefanklut/new_augments

stefanklut · Apr 24, 2024 · a24d178 · a24d178
2 parents 11b14b5 + ad6da58
commit a24d178
Show file tree

Hide file tree

Showing 5 changed files with 465 additions and 42 deletions.
diff --git a/configs/extra_defaults.py b/configs/extra_defaults.py
@@ -93,6 +93,23 @@
 _C.INPUT.GRAYSCALE = CN()
 _C.INPUT.GRAYSCALE.PROBABILITY = 0.5
 
+_C.INPUT.ADAPTIVE_THESHOLDING = CN()
+_C.INPUT.ADAPTIVE_THESHOLDING.PROBABILITY = 0.5
+
+_C.INPUT.INVERT = CN()
+_C.INPUT.INVERT.PROBABILITY = 0.5
+_C.INPUT.INVERT.MAX_VALUE = 255
+
+_C.JPEG_COMPRESSION = CN()
+_C.JPEG_COMPRESSION.PROBABILITY = 0.5
+_C.JPEG_COMPRESSION.MIN_QUALITY = 0
+_C.JPEG_COMPRESSION.MAX_QUALITY = 100
+
+_C.INPUT.NOISE = CN()
+_C.INPUT.NOISE.PROBABILITY = 0.5
+_C.INPUT.NOISE.MIN_STD = 0
+_C.INPUT.NOISE.MAX_STD = 50
+
 _C.INPUT.BRIGHTNESS = CN()
 _C.INPUT.BRIGHTNESS.PROBABILITY = 0.5
 _C.INPUT.BRIGHTNESS.MIN_INTENSITY = 0.5
@@ -108,6 +125,11 @@
 _C.INPUT.SATURATION.MIN_INTENSITY = 0.5
 _C.INPUT.SATURATION.MAX_INTENSITY = 1.5
 
+_C.INPUT.HUE = CN()
+_C.INPUT.HUE.PROBABILITY = 0.5
+_C.INPUT.HUE.MIN_DELTA = -0.5
+_C.INPUT.HUE.MAX_DELTA = 0.5
+
 _C.INPUT.GAUSSIAN_FILTER = CN()
 _C.INPUT.GAUSSIAN_FILTER.PROBABILITY = 0.5
 _C.INPUT.GAUSSIAN_FILTER.MIN_SIGMA = 0.0

diff --git a/configs/segmentation/baseline/baseline_general.yaml b/configs/segmentation/baseline/baseline_general.yaml
@@ -34,23 +34,44 @@ INPUT:
   GRAYSCALE:
     PROBABILITY: 0.1
 
+  ADAPTIVE_THRESHOLDING:
+    PROBABILITY: 0.05
+
   BRIGHTNESS:
-    PROBABILITY: 0.2
+    PROBABILITY: 0.25
     MIN_INTENSITY: 0.5
     MAX_INTENSITY: 1.5
 
   CONTRAST:
-    PROBABILITY: 0.2
+    PROBABILITY: 0.25
     MIN_INTENSITY: 0.5
     MAX_INTENSITY: 1.5
 
   SATURATION:
-    PROBABILITY: 0.1
+    PROBABILITY: 0.25
     MIN_INTENSITY: 0.5
     MAX_INTENSITY: 1.5
 
+  HUE:
+    PROBABILITY: 0.25
+    MIN_DELTA: -0.25
+    MAX_DELTA: 0.25
+
+  INVERT:
+    PROBABILITY: 0.1
+
+  JPEG_COMPRESSION:
+    PROBABILITY: 0.25
+    MIN_QUALITY: 50
+    MAX_QUALITY: 100
+
+  NOISE:
+    PROBABILITY: 0.25
+    MIN_STD: 10
+    MAX_STD: 32
+
   GAUSSIAN_FILTER:
-    PROBABILITY: 0.2
+    PROBABILITY: 0.25
     MIN_SIGMA: 0.5
     MAX_SIGMA: 1.5
 
@@ -61,7 +82,7 @@ INPUT:
     PROBABILITY: 0.
 
   ELASTIC_DEFORMATION:
-    PROBABILITY: 0.2
+    PROBABILITY: 0.25
     ALPHA: 0.1
     SIGMA: 0.01
 

diff --git a/configs/segmentation/segmentation_base.yaml b/configs/segmentation/segmentation_base.yaml
@@ -61,6 +61,22 @@ INPUT:
   GRAYSCALE:
     PROBABILITY: 0.
 
+  ADAPTIVE_THRESHOLDING:
+    PROBABILITY: 0.
+
+  INVERT:
+    PROBABILITY: 0.
+
+  JPEG_COMPRESSION:
+    PROBABILITY: 0.
+    MIN_QUALITY: 50
+    MAX_QUALITY: 100
+
+  NOISE:
+    PROBABILITY: 0.
+    MIN_STD: 10
+    MAX_STD: 32
+
   BRIGHTNESS:
     PROBABILITY: 0.
     MIN_INTENSITY: 0.5
@@ -76,6 +92,11 @@ INPUT:
     MIN_INTENSITY: 0.5
     MAX_INTENSITY: 1.5
 
+  HUE:
+    PROBABILITY: 0.
+    MIN_DELTA: -0.25
+    MAX_DELTA: 0.25
+
   GAUSSIAN_FILTER:
     PROBABILITY: 0.
     MIN_SIGMA: 0.5
@@ -111,6 +132,10 @@ INPUT:
       PROBABILITY: 0.5
       STANDARD_DEVIATION: 0.12
 
+  ORIENTATION:
+    PROBABILITY: 0.
+    PERCENTAGES: (0.25, 0.25, 0.25, 0.25)
+
 TEST:
   EVAL_PERIOD: 10000