diff --git a/configs/extra_defaults.py b/configs/extra_defaults.py index 87d276f..3fabad3 100644 --- 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 index 237ee15..28c5def 100644 --- 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