virtual-tryon-demo

Running on Zero

File size: 523 Bytes

d4f8fc2

_BASE_: "cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv.yaml"
MODEL:
  MASK_ON: True
  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: ("demo_train",)
  TEST: ("demo_val",)
OUTPUT_DIR: "../../data/DemoDataset/detectron2_prediction"