photo config

stefanklut · Nov 22, 2023 · 482da62 · 482da62
1 parent 43f1ead
commit 482da62
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/configs/segmentation/region/region_dataset_photo.yaml b/configs/segmentation/region/region_dataset_photo.yaml
@@ -6,18 +6,18 @@ NAME: photo
 
 PREPROCESS:
   REGION:
-    REGIONS: ["Text", "Photo", "Drawing"]
+    REGIONS: ["Photo"]
     MERGE_REGIONS: []
-    REGION_TYPE: []
+    REGION_TYPE: ["ImageRegion:Photo"]
 
 MODEL:
   MODE: region
 
   SEM_SEG_HEAD:
-    NUM_CLASSES: 4
+    NUM_CLASSES: 2
 
   ROI_HEADS:
-    NUM_CLASSES: 3
+    NUM_CLASSES: 1
 
   WEIGHTS:
 
@@ -80,12 +80,12 @@ TRAIN:
 
 TEST:
   WEIGHTS:
-  EVAL_PERIOD: 1000
+  EVAL_PERIOD: 10000
 
 SOLVER:
-  IMS_PER_BATCH: 2
-  CHECKPOINT_PERIOD: 2000
+  IMS_PER_BATCH: 8
+  CHECKPOINT_PERIOD: 25000
   BASE_LR: 0.0002
   GAMMA: 0.1
   STEPS: () #(80000, 120000, 160000)
-  MAX_ITER: 10000
+  MAX_ITER: 250000