config.json

{
  "model_type": "yolov8",
  "num_labels": 2,
  "image_size": 640,
  "backbone": "CSPDarknet53", 
  "pretrained": true,
  "input_channels": 3, 
  "conf_thresh": 0.5, 
  "iou_thresh": 0.4,
  "max_detections": 100,
  "stride": [8, 16, 32],
  "anchor_generator": {
    "sizes": [[12, 16, 32, 64, 128], [256, 512, 1024]],
    "aspect_ratios": [[1, 2, 0.5], [1, 2, 0.5]]
  },
  "nms_thresh": 0.4
}