16 lines
316 B
YAML
16 lines
316 B
YAML
|
_BASE_: "../Base-RCNN-FPN.yaml"
|
||
|
MODEL:
|
||
|
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
||
|
MASK_ON: True
|
||
|
RESNETS:
|
||
|
DEPTH: 50
|
||
|
ROI_HEADS:
|
||
|
NAME: CascadeROIHeads
|
||
|
ROI_BOX_HEAD:
|
||
|
CLS_AGNOSTIC_BBOX_REG: True
|
||
|
RPN:
|
||
|
POST_NMS_TOPK_TRAIN: 2000
|
||
|
SOLVER:
|
||
|
STEPS: (210000, 250000)
|
||
|
MAX_ITER: 270000
|