File size: 576 Bytes
63694c6 |
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 28 29 30 31 32 33 34 |
_BASE_: "Base-RCNN-FPN.yaml"
MODEL:
WEIGHTS: ""
MASK_ON: False
RESNETS:
DEPTH: 50
STRIDE_IN_1X1: False
NUM_GROUPS: 32
WIDTH_PER_GROUP: 4
NORM: "GN"
FPN:
NORM: "GN"
ROI_HEADS:
NAME: "CascadeROIHeads"
NUM_CLASSES: 5
SCORE_THRESH_TEST: 0.1
NMS_THRESH_TEST: 0.001
ROI_BOX_HEAD:
CLS_AGNOSTIC_BBOX_REG: True
PIXEL_MEAN:
- 238.234
- 238.14
- 238.145
PIXEL_STD:
- 7.961
- 7.876
- 7.81
INPUT:
MIN_SIZE_TEST: 800
MIN_SIZE_TRAIN: !!python/tuple
- 800
- 1200
NMS_THRESH_CLASS_AGNOSTIC: 0.001
|