Spaces:
Running
on
Zero
Running
on
Zero
Virtual-Try-On
/
preprocess
/humanparsing
/mhp_extension
/detectron2
/configs
/LVIS-InstanceSegmentation
/mask_rcnn_R_101_FPN_1x.yaml
_BASE_: "../Base-RCNN-FPN.yaml" | |
MODEL: | |
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl" | |
MASK_ON: True | |
RESNETS: | |
DEPTH: 101 | |
ROI_HEADS: | |
NUM_CLASSES: 1230 | |
SCORE_THRESH_TEST: 0.0001 | |
INPUT: | |
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800) | |
DATASETS: | |
TRAIN: ("lvis_v0.5_train",) | |
TEST: ("lvis_v0.5_val",) | |
TEST: | |
DETECTIONS_PER_IMAGE: 300 # LVIS allows up to 300 | |
DATALOADER: | |
SAMPLER_TRAIN: "RepeatFactorTrainingSampler" | |
REPEAT_THRESHOLD: 0.001 | |