From 195badf3ad3c5623360dc83c68694c31ef8e81a4 Mon Sep 17 00:00:00 2001 From: Stefan Klut Date: Tue, 9 Apr 2024 12:26:35 +0200 Subject: [PATCH] Naming convention is incorrect --- datasets/augmentations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datasets/augmentations.py b/datasets/augmentations.py index 9bd30f3..7387655 100644 --- a/datasets/augmentations.py +++ b/datasets/augmentations.py @@ -1036,7 +1036,7 @@ def build_augmentation(cfg: CfgNode, mode: str = "train") -> list[T.Augmentation elif cfg.PREPROCESS.RESIZE.RESIZE_MODE in ["shortest_edge", "longest_edge"]: min_size = cfg.PREPROCESS.RESIZE.MIN_SIZE max_size = cfg.PREPROCESS.RESIZE.MAX_SIZE - sample_style = cfg.PREPROCESS.RESIZE.MIN_SIZE_SAMPLING + sample_style = cfg.PREPROCESS.RESIZE.RESIZE_SAMPLING if cfg.PREPROCESS.RESIZE.RESIZE_MODE == "shortest_edge": augmentation.append(ResizeShortestEdge(min_size, max_size, sample_style)) elif cfg.PREPROCESS.RESIZE.RESIZE_MODE == "longest_edge":