{
  "architecture": "RFDETRSmall",
  "framework": "PyTorch",
  "input_resolution": 576,
  "classes": [
    {
      "id": 0,
      "name": "helmet"
    }
  ],
  "recommended_detection_threshold": 0.35,
  "checkpoint": {
    "filename": "helmet_only_576_best.pth",
    "size_bytes": 127830863,
    "sha256": "0e6741285fe7bf2a19cd18b37e558cf98e8ee1730dbb4f591ee0b2b75fb8568e"
  },
  "training": {
    "seed": 42,
    "selected_epoch": 19,
    "completed_epochs": 20,
    "global_steps": 2000
  },
  "test_metrics": {
    "images": 1234,
    "ground_truth_helmets": 597,
    "map_50_95": 0.5890050576948107,
    "map_50": 0.8961793948081103,
    "map_75": 0.6740506908162793,
    "mar_100": 0.7408710217755444,
    "precision_50": 0.8667736757624398,
    "recall_50": 0.9045226130653267,
    "f1_50": 0.8852459016393442,
    "confidence_at_best_f1": 0.5697923302650452
  },
  "source_repository": "https://github.com/matheeusper/epi-with-keypoints",
  "source_dataset": {
    "repo_id": "51ddhesh/PPE_Detection",
    "url": "https://huggingface.co/datasets/51ddhesh/PPE_Detection",
    "license": "CC-BY-4.0",
    "classes": [
      "Gloves",
      "Vest",
      "goggles",
      "helmet",
      "mask",
      "safety_shoe"
    ],
    "splits": {
      "train_images": 8774,
      "validation_images": 2070,
      "test_images": 1234
    }
  }
}