boxsnake_R_101_FPN_1x.yaml

_BASE_: "../Base-BoxSnake-RCNN-FPN.yaml"
MODEL:
  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
  MASK_ON: True
  ROI_MASK_HEAD:
    NAME: "PolygonHead"
    POOLER_TYPE: ""
  POLYGON_HEAD:
    IN_FEATURES: ["p2", "p3", "p4", "p5"]
    PRED_WITHIN_BOX: False
    POLY_NUM_PTS: 64
    CLS_AGNOSTIC_MASK: True
    PREPOOL: False
    UPSAMPLING: False
  DIFFRAS:
    RESOLUTIONS: (64, 64, 64, 64, 64, 64, 64, 64)
    USE_RASTERIZED_GT: False
    INV_SMOOTHNESS_SCHED: (0.1,)
  RESNETS:
    DEPTH: 50
  ROI_HEADS:
    PROPOSAL_ONLY_GT: False
  BOX_SUP:
    ENABLE: True
    LOSS_POINTS_PROJ: True
    LOSS_POINTS_PROJ_WEIGHT: 1.0
    LOSS_LOCAL_PAIRWISE: True
    LOSS_PAIRWISE_WARMUP_ITER: 10000
    LOCAL_PAIRWISE_KERNEL_SIZE: 3
    LOCAL_PAIRWISE_DILATION: 2
    LOSS_LOCAL_PAIRWISE_WEIGHT: 0.5
    LOSS_GLOBAL_PAIRWISE: True
    LOSS_GLOBAL_PAIRWISE_WEIGHT: 0.03
    CROP_PREDICTS: True
    CROP_SIZE: 64
    MASK_PADDING_SIZE: 4
SOLVER:
  OPTIMIZER: "ADAM"
  BASE_LR: 1e-4
  WEIGHT_DECAY: 0.1
  WEIGHT_DECAY_NORM: 0.0
  STEPS: (60000, 80000)
  MAX_ITER: 90000
  IMS_PER_BATCH: 16
  CLIP_GRADIENTS:
    ENABLED: False
DATASETS:
  TRAIN: ("coco_2017_train",)
  TEST: ("coco_2017_val",)
VIS_PERIOD: 0
TEST:
  EVAL_PERIOD: 10000