Virtual-Try-On

Running on Zero

File size: 547 Bytes

938e515

_BASE_: "cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv.yaml"
MODEL:
  MASK_ON: True
  WEIGHTS: "./finetune_ouput/model_final.pth"
  ROI_HEADS:
    NMS_THRESH_TEST: 0.95
    SCORE_THRESH_TEST: 0.5
    NUM_CLASSES: 1
SOLVER:
  IMS_PER_BATCH: 1
  STEPS: (30000, 45000)
  MAX_ITER: 50000
  BASE_LR: 0.02
INPUT:
  MIN_SIZE_TRAIN: (640, 864)
  MIN_SIZE_TRAIN_SAMPLING: "range"
  MAX_SIZE_TRAIN: 1440
  CROP:
    ENABLED: True
TEST:
  AUG:
    ENABLED: True
DATASETS:
  TRAIN: ("CIHP_trainval",)
  TEST: ("CIHP_test",)
OUTPUT_DIR: "./inference_output"