Spaces:
Runtime error
Runtime error
fifa-tryon-demo
/
Self-Correction-Human-Parsing-for-ACGPN
/mhp_extension
/detectron2
/configs
/Misc
/cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv.yaml
_BASE_: "../Base-RCNN-FPN.yaml" | |
MODEL: | |
MASK_ON: True | |
WEIGHTS: "catalog://ImageNetPretrained/FAIR/X-152-32x8d-IN5k" | |
RESNETS: | |
STRIDE_IN_1X1: False # this is a C2 model | |
NUM_GROUPS: 32 | |
WIDTH_PER_GROUP: 8 | |
DEPTH: 152 | |
DEFORM_ON_PER_STAGE: [False, True, True, True] | |
ROI_HEADS: | |
NAME: "CascadeROIHeads" | |
ROI_BOX_HEAD: | |
NAME: "FastRCNNConvFCHead" | |
NUM_CONV: 4 | |
NUM_FC: 1 | |
NORM: "GN" | |
CLS_AGNOSTIC_BBOX_REG: True | |
ROI_MASK_HEAD: | |
NUM_CONV: 8 | |
NORM: "GN" | |
RPN: | |
POST_NMS_TOPK_TRAIN: 2000 | |
SOLVER: | |
IMS_PER_BATCH: 128 | |
STEPS: (35000, 45000) | |
MAX_ITER: 50000 | |
BASE_LR: 0.16 | |
INPUT: | |
MIN_SIZE_TRAIN: (640, 864) | |
MIN_SIZE_TRAIN_SAMPLING: "range" | |
MAX_SIZE_TRAIN: 1440 | |
CROP: | |
ENABLED: True | |
TEST: | |
EVAL_PERIOD: 2500 | |