JaMe76's picture
Upload CASCADE_RCNN_R_50_FPN_GN.yaml
63694c6
raw history blame
No virus
576 Bytes
_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