CenterNet2_R2-101-DCN-BiFPN_1280_4x.yaml

_BASE_: "Base-CenterNet2.yaml"
MODEL:
  BACKBONE:
    NAME: "build_res2net_bifpn_backbone"
  BIFPN:
    NUM_BIFPN: 7
    OUT_CHANNELS: 288
  WEIGHTS: "output/r2_101.pkl"
  RESNETS:
    DEPTH: 101
    WIDTH_PER_GROUP: 26
    DEFORM_ON_PER_STAGE: [False, False, True, True] # on Res4, Res5
    DEFORM_MODULATED: True
  PIXEL_MEAN: [123.675, 116.280, 103.530]
  PIXEL_STD: [58.395, 57.12, 57.375]
  CENTERNET:
    USE_DEFORMABLE: True
  ROI_HEADS:
    IN_FEATURES: ["p3", "p4"]
INPUT:
  FORMAT: RGB
TEST:
  EVAL_PERIOD: 7500
SOLVER:
  MAX_ITER: 180000
  CHECKPOINT_PERIOD: 60000
  LR_SCHEDULER_NAME: "WarmupCosineLR"
  BASE_LR: 0.04
  IMS_PER_BATCH: 32
INPUT:
  CUSTOM_AUG: EfficientDetResizeCrop
  TRAIN_SIZE: 1280