_BASE_: "Base-RCNN-FPN-MC.yaml"
MODEL:
  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
  RESNETS:
    DEPTH: 101
  DENSEPOSE_ON: True
  ROI_HEADS:
    NAME: "DensePoseROIHeads"
    IN_FEATURES: ["p2", "p3", "p4", "p5"]
    NUM_CLASSES: 1
  ROI_DENSEPOSE_HEAD:
    NAME: "DensePoseDeepLabHead"
    UV_CONFIDENCE:
      ENABLED: True
      TYPE: "iid_iso"
    POINT_REGRESSION_WEIGHTS: 0.0005
    POOLER_TYPE: "ROIAlign"
    NUM_COARSE_SEGM_CHANNELS: 2
    COARSE_SEGM_TRAINED_BY_MASKS: True
    INDEX_WEIGHTS: 1.0
SOLVER:
  CLIP_GRADIENTS:
    ENABLED: True
  WARMUP_FACTOR: 0.025
DATASETS:
  TRAIN: ("base_coco_2017_train", "densepose_coco_2014_train")
  TEST: ("densepose_chimps",)
