File size: 787 Bytes
2111dbf 7f95b4c 2a4ac51 2111dbf ffbfcf5 |
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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
_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: 2
SCORE_THRESH_TEST: 0.4
NMS_THRESH_TEST: 0.001
ROI_BOX_HEAD:
CLS_AGNOSTIC_BBOX_REG: True
PIXEL_MEAN:
- 238.21
- 238.22
- 238.32
PIXEL_STD:
- 9.14
- 8.569
- 8.75
ANCHOR_GENERATOR:
ASPECT_RATIOS:
- - 0.1
- 0.5
- 1.5
SIZES:
- - 16
- - 32
- - 64
- - 128
- - 256
INPUT:
MIN_SIZE_TEST: 800
MAX_SIZE_TEST: 1400
MIN_SIZE_TRAIN: !!python/tuple
- 200
- 200
TEST:
DETECTIONS_PER_IMAGE: 350
NMS_THRESH_CLASS_AGNOSTIC: 0.999
|