typo stopped training

stefanklut · Apr 29, 2024 · da2aba0 · da2aba0
1 parent a24d178
commit da2aba0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/configs/extra_defaults.py b/configs/extra_defaults.py
@@ -93,17 +93,17 @@
 _C.INPUT.GRAYSCALE = CN()
 _C.INPUT.GRAYSCALE.PROBABILITY = 0.5
 
-_C.INPUT.ADAPTIVE_THESHOLDING = CN()
-_C.INPUT.ADAPTIVE_THESHOLDING.PROBABILITY = 0.5
+_C.INPUT.ADAPTIVE_THRESHOLDING = CN()
+_C.INPUT.ADAPTIVE_THRESHOLDING.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.JPEG_COMPRESSION = CN()
+_C.INPUT.JPEG_COMPRESSION.PROBABILITY = 0.5
+_C.INPUT.JPEG_COMPRESSION.MIN_QUALITY = 0
+_C.INPUT.JPEG_COMPRESSION.MAX_QUALITY = 100
 
 _C.INPUT.NOISE = CN()
 _C.INPUT.NOISE.PROBABILITY = 0.5

diff --git a/datasets/augmentations.py b/datasets/augmentations.py
@@ -1168,8 +1168,8 @@ def build_augmentation(cfg: CfgNode, mode: str = "train") -> list[T.Augmentation
                 max_size = cfg.INPUT.MAX_SIZE_TRAIN
                 sample_style = cfg.INPUT.MIN_SIZE_TRAIN_SAMPLING
             elif mode == "val":
-                min_size = cfg.INPUT.MIN_SIZE_TRAIN
-                max_size = cfg.INPUT.MAX_SIZE_TRAIN
+                min_size = cfg.INPUT.MIN_SIZE_TEST
+                max_size = cfg.INPUT.MAX_SIZE_TEST
                 sample_style = "choice"
             elif mode == "test":
                 min_size = cfg.INPUT.MIN_SIZE_TEST