OOTDiffusion-VirtualTryOnClothing
/
preprocess
/humanparsing
/mhp_extension
/detectron2
/configs
/Base-RCNN-C4.yaml
MODEL: | |
META_ARCHITECTURE: "GeneralizedRCNN" | |
RPN: | |
PRE_NMS_TOPK_TEST: 6000 | |
POST_NMS_TOPK_TEST: 1000 | |
ROI_HEADS: | |
NAME: "Res5ROIHeads" | |
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 | |