config.json

{
  "backbone_arch": "P5",
  "backbone_channel_attention": true,
  "backbone_deepen_factor": 1.0,
  "backbone_expand_ratio": 0.5,
  "backbone_widen_factor": 1.0,
  "head_exp_on_reg": true,
  "head_feat_channels": 256,
  "head_in_channels": 256,
  "head_pred_kernel_size": 1,
  "head_share_conv": true,
  "head_stacked_convs": 2,
  "head_with_objectness": false,
  "input_size": [
    640,
    640
  ],
  "max_detections": 100,
  "model_type": "rtmdet",
  "neck_expand_ratio": 0.5,
  "neck_in_channels": [
    256,
    512,
    1024
  ],
  "neck_num_csp_blocks": 3,
  "neck_out_channels": 256,
  "nms_threshold": 0.6,
  "num_classes": 80,
  "score_threshold": 0.05,
  "strides": [
    8,
    16,
    32
  ],
  "transformers_version": "5.2.0",
  "auto_map": {
    "AutoConfig": "configuration_rtmdet.RTMDetConfig",
    "AutoModel": "modeling_rtmdet.RTMDetModel"
  }
}