OOTDiffusion-VirtualTryOnClothing
/
preprocess
/humanparsing
/mhp_extension
/detectron2
/configs
/Base-RCNN-DilatedC5.yaml
MODEL: | |
META_ARCHITECTURE: "GeneralizedRCNN" | |
RESNETS: | |
OUT_FEATURES: ["res5"] | |
RES5_DILATION: 2 | |
RPN: | |
IN_FEATURES: ["res5"] | |
PRE_NMS_TOPK_TEST: 6000 | |
POST_NMS_TOPK_TEST: 1000 | |
ROI_HEADS: | |
NAME: "StandardROIHeads" | |
IN_FEATURES: ["res5"] | |
ROI_BOX_HEAD: | |
NAME: "FastRCNNConvFCHead" | |
NUM_FC: 2 | |
POOLER_RESOLUTION: 7 | |
ROI_MASK_HEAD: | |
NAME: "MaskRCNNConvUpsampleHead" | |
NUM_CONV: 4 | |
POOLER_RESOLUTION: 14 | |
DATASETS: | |
TRAIN: ("coco_2017_train",) | |
TEST: ("coco_2017_val",) | |
SOLVER: | |
IMS_PER_BATCH: 16 | |
BASE_LR: 0.02 | |
STEPS: (60000, 80000) | |
MAX_ITER: 90000 | |
INPUT: | |
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800) | |
VERSION: 2 | |