resnet.yaml

_BASE_: "Base-RCNN-FPN.yaml"
MODEL:
  META_ARCHITECTURE: "GeneralizedRCNN"
  WEIGHTS: "model_final.pth"
  #WEIGHTS: "./data/SPEED-Detection/faster-rcnn/vanilla/random_seed_0/model_final.pth"

  MASK_ON: False
  RESNETS:
    DEPTH: 50
  ROI_HEADS:
    NAME: "StandardROIHeads"
    NUM_CLASSES: 9
# INPUT:
#   MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800)
#   MIN_SIZE_TEST: 800
DATASETS:
  TRAIN: ('speed_custom_train',)
  TEST: ('speed_custom_val',)
SOLVER:
  IMS_PER_BATCH: 18
  BASE_LR: 0.01
  STEPS: (2800, 3700)
  MAX_ITER: 4200
  WARMUP_ITERS: 50
  CHECKPOINT_PERIOD : 2100
TEST:
  EVAL_PERIOD: 1000