Spaces:
Sleeping
Sleeping
File size: 329 Bytes
02d450e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
# Example Detectron2 config for parts model (5 classes)
_BASE_: "COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"
MODEL:
ROI_HEADS:
NUM_CLASSES: 5
DATASETS:
TRAIN: ("parts_train",)
TEST: ("parts_val",)
SOLVER:
IMS_PER_BATCH: 2
BASE_LR: 0.00025
MAX_ITER: 3000
OUTPUT_DIR: "./weights/parts_model"
|