Spaces:
Running
on
Zero
Running
on
Zero
Virtual-Try-On
/
preprocess
/humanparsing
/mhp_extension
/detectron2
/configs
/Misc
/cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv_parsing.yaml
_BASE_: "../Base-RCNN-FPN.yaml" | |
MODEL: | |
MASK_ON: True | |
# WEIGHTS: "catalog://ImageNetPretrained/FAIR/X-152-32x8d-IN5k" | |
WEIGHTS: "model_0039999_e76410.pkl" | |
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" | |
NUM_CLASSES: 1 | |
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 | |
IMS_PER_BATCH: 1 | |
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 | |
DATASETS: | |
TRAIN: ("CIHP_train","VIP_trainval") | |
TEST: ("CIHP_val",) | |