26 lines
523 B
YAML
26 lines
523 B
YAML
|
_BASE_: "cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv.yaml"
|
||
|
|
MODEL:
|
||
|
|
MASK_ON: True
|
||
|
|
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: ("demo_train",)
|
||
|
|
TEST: ("demo_val",)
|
||
|
|
OUTPUT_DIR: "../../data/DemoDataset/detectron2_prediction"
|