File size: 547 Bytes
5a486d6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
_BASE_: "cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv.yaml"
MODEL:
MASK_ON: True
WEIGHTS: "./finetune_ouput/model_final.pth"
ROI_HEADS:
NMS_THRESH_TEST: 0.95
SCORE_THRESH_TEST: 0.5
NUM_CLASSES: 1
SOLVER:
IMS_PER_BATCH: 1
STEPS: (30000, 45000)
MAX_ITER: 50000
BASE_LR: 0.02
INPUT:
MIN_SIZE_TRAIN: (640, 864)
MIN_SIZE_TRAIN_SAMPLING: "range"
MAX_SIZE_TRAIN: 1440
CROP:
ENABLED: True
TEST:
AUG:
ENABLED: True
DATASETS:
TRAIN: ("CIHP_trainval",)
TEST: ("CIHP_test",)
OUTPUT_DIR: "./inference_output"
|