config.json

{
  "framework": "pytorch",
  "type": "detector",
  "classes": [
    "bicycle_rider"
  ],
  "pytorch": {
    "model_file": "best.pt",
    "input_size": [640, 640],
    "confidence_threshold": 0.5,
    "nms_threshold": 0.45
  }
}