From 482da62d601494c1566470621c3ed9a9cd09d8e3 Mon Sep 17 00:00:00 2001 From: Stefan Klut Date: Wed, 22 Nov 2023 12:19:15 +0100 Subject: [PATCH] photo config --- .../region/region_dataset_photo.yaml | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) 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