diff --git a/configs/segmentation/region/region_dataset_photo.yaml b/configs/segmentation/region/region_dataset_photo.yaml index 9aea631..c6c4fbb 100644 --- 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