Spaces:
Runtime error
Runtime error
IDM-VTON
/
preprocess
/humanparsing
/mhp_extension
/detectron2
/configs
/PascalVOC-Detection
/faster_rcnn_R_50_C4.yaml
_BASE_: "../Base-RCNN-C4.yaml" | |
MODEL: | |
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl" | |
MASK_ON: False | |
RESNETS: | |
DEPTH: 50 | |
ROI_HEADS: | |
NUM_CLASSES: 20 | |
INPUT: | |
MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800) | |
MIN_SIZE_TEST: 800 | |
DATASETS: | |
TRAIN: ('voc_2007_trainval', 'voc_2012_trainval') | |
TEST: ('voc_2007_test',) | |
SOLVER: | |
STEPS: (12000, 16000) | |
MAX_ITER: 18000 # 17.4 epochs | |
WARMUP_ITERS: 100 | |