gsnet

_BASE_: "../Base-RCNN-FPN.yaml"
MODEL:
  KEYPOINT_ON: True
  ROI_HEADS:
    NUM_CLASSES: 79
  ROI_BOX_HEAD:
    SMOOTH_L1_BETA: 0.5  # Keypoint AP degrades (though box AP improves) when using plain L1 loss
  RPN:
    # Detectron1 uses 2000 proposals per-batch, but this option is per-image in detectron2.
    # 1000 proposals per-image is found to hurt box AP.
    # Therefore we increase it to 1500 per-image.
    POST_NMS_TOPK_TRAIN: 1500
DATASETS:
  TRAIN: ("apollo_train",)
  TEST: ("apollo_val",)
INPUT:
  #MIN_SIZE_TRAIN: (1200, 1260, 1320, 1380, 1440, 1500)
  MIN_SIZE_TRAIN: (1500,)
  #MIN_SIZE_TRAIN: (2710,)
  MAX_SIZE_TRAIN: 3384
  #MIN_SIZE_TEST: 2710
  MIN_SIZE_TEST: 1500
  MAX_SIZE_TEST: 3384