config.jsonl

{
  "ANCHOR_GENERATOR": {
    "ANGLES": [
      [
        -90.0,
        0.0,
        90.0
      ]
    ],
    "ASPECT_RATIOS": [
      [
        0.5,
        1.0,
        2.0
      ]
    ],
    "OFFSET": 0,
    "SIZES": [
      [
        64,
        128,
        256,
        512
      ]
    ]
  },
  "BACKBONE": {
    "FREEZE_AT": 2
  },
  "INPUT": {
    "FORMAT": "BGR",
    "MAX_SIZE_TEST": 1333,
    "MAX_SIZE_TRAIN": 1333,
    "MIN_SIZE_TEST": 800,
    "MIN_SIZE_TRAIN": 640
  },
  "MAX_DETECTIONS": 36,
  "MIN_DETECTIONS": 36,
  "MODEL": {
    "CHKPOINT": "",
    "DEVICE": "cpu",
    "LOAD_PROPOSALS": false,
    "MAX_POOL": true,
    "PIXEL_MEAN": [
      102.9801,
      115.9465,
      122.7717
    ],
    "PIXEL_STD": [
      1.0,
      1.0,
      1.0
    ]
  },
  "PAD_VALUE": 0.0,
  "PROPOSAL_GENERATOR": {
    "HIDDEN_CHANNELS": 512,
    "MIN_SIZE": 0.0
  },
  "RESNETS": {
    "DEPTH": 101,
    "NORM": "BN",
    "NUM_GROUPS": 1,
    "OUT_FEATURES": [
      "res4"
    ],
    "RES2_OUT_CHANNELS": 256,
    "RES5_DILATION": 1,
    "STEM_OUT_CHANNELS": 64,
    "STRIDE_IN_1X1": true,
    "WIDTH_PER_GROUP": 64
  },
  "ROI_BOX_HEAD": {
    "ATTR": true,
    "BBOX_REG_WEIGHTS": [
      10.0,
      10.0,
      5.0,
      5.0
    ],
    "CLS_AGNOSTIC_BBOX_REG": false,
    "CONV_DIM": 256,
    "FC_DIM": 1024,
    "NUM_ATTRS": 400,
    "NUM_CONV": 0,
    "NUM_FC": 0,
    "POOLER_RESOLUTION": 14,
    "POOLER_SAMPLING_RATIO": 0,
    "RES5HALVE": false,
    "SMOOTH_L1_BETA": 0.0
  },
  "ROI_HEADS": {
    "IN_FEATURES": [
      "res4"
    ],
    "IOU_LABELS": [
      0,
      1
    ],
    "IOU_THRESHOLDS": [
      0.5
    ],
    "NMS_THRESH_TEST": [
      0.5,
      0.6,
      0.7
    ],
    "NUM_CLASSES": 1600,
    "POSITIVE_FRACTION": 0.25,
    "PROPOSAL_APPEND_GT": true,
    "SCORE_THRESH_TEST": 0.2
  },
  "RPN": {
    "BATCH_SIZE_PER_IMAGE": 1,
    "BBOX_REG_WEIGHTS": [
      1.0,
      1.0,
      1.0,
      1.0
    ],
    "BOUNDARY_THRESH": -1.0,
    "IN_FEATURES": [
      "res4"
    ],
    "IOU_LABELS": [
      0.0,
      -1.0,
      1.0
    ],
    "IOU_THRESHOLDS": [
      0.3,
      0.7
    ],
    "LOSS_WEIGHT": 1.0,
    "NMS_THRESH": 0.7,
    "POSITIVE_FRACTION": 0.5,
    "POST_NMS_TOPK_TEST": 300,
    "POST_NMS_TOPK_TRAIN": 2000,
    "PRE_NMS_TOPK_TEST": 6000,
    "PRE_NMS_TOPK_TRAIN": 12000,
    "SMOOTH_L1_BETA": 0.0
  },
  "SIZE_DIVISIBILITY": 0,
  "SOLVER": {
    "BASE_LR": 0.002,
    "BIAS_LR_FACTOR": 1.0,
    "CHECKPOINT_PERIOD": 5000,
    "GAMMA": 0.1,
    "IMS_PER_BATCH": 1.0,
    "LR_SCHEDULER_NAME": "warmupmultisteplr",
    "MAX_ITER": 270000,
    "MOMENTUM": 0.9,
    "STEPS": [
      210000,
      250000
    ],
    "WARMUP_FACTOR": 0.001,
    "WARMUP_ITERS": 1000,
    "WARMUP_METHOD": "linear",
    "WEIGHT_DECAY": 0.0001,
    "WEIGHT_DECAY_BIAS": 0.0,
    "WEIGHT_DECAY_NORM": 0.0
  }
}