set MIN_SIZE_TRAIN to 1024

stefanklut · Dec 4, 2023 · 1b7c3db · 1b7c3db
1 parent d56c3fb
commit 1b7c3db
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/configs/segmentation/baseline/baseline_ovdr2.yaml b/configs/segmentation/baseline/baseline_ovdr2.yaml
@@ -18,7 +18,7 @@ PREPROCESS:
 INPUT:
   RESIZE_MODE: "shortest_edge"
   MIN_SIZE_TRAIN_SAMPLING: choice
-  MIN_SIZE_TRAIN: (896, 960, 1024, 1088, 1152)
+  MIN_SIZE_TRAIN: (1024,)
   MAX_SIZE_TRAIN: 2024
 
   MAX_SIZE_TEST: 2024

diff --git a/configs/segmentation/baseline/baseline_ovdr3.yaml b/configs/segmentation/baseline/baseline_ovdr3.yaml
@@ -18,7 +18,7 @@ PREPROCESS:
 INPUT:
   RESIZE_MODE: "shortest_edge"
   MIN_SIZE_TRAIN_SAMPLING: choice
-  MIN_SIZE_TRAIN: (896, 960, 1024, 1088, 1152)
+  MIN_SIZE_TRAIN: (1024,)
   MAX_SIZE_TRAIN: 2024
 
   MAX_SIZE_TEST: 2024

diff --git a/configs/segmentation/segmentation_base.yaml b/configs/segmentation/segmentation_base.yaml
@@ -45,7 +45,7 @@ PREPROCESS:
 INPUT:
   RESIZE_MODE: "shortest_edge"
   MIN_SIZE_TRAIN_SAMPLING: choice
-  MIN_SIZE_TRAIN: (896, 960, 1024, 1088, 1152)
+  MIN_SIZE_TRAIN: (1024,)
   MAX_SIZE_TRAIN: 2048
   SCALING_TRAIN: 0.5