config.json

{
  "architectures": [
    "YolosForObjectDetection"
  ],
  "model_type": "yolos",
  "input_size": [640, 640],
  "anchors": [
    [10, 13],
    [16, 30],
    [33, 23]
  ],
  "classes": ["0", "1"],
  "output_layers": [
    {"layer_size": [52, 52], "anchors": 3},
    {"layer_size": [26, 26], "anchors": 3},
    {"layer_size": [13, 13], "anchors": 3}
  ],
  "nms_threshold": 0.5
}