File size: 1,342 Bytes
f288ae6 9e6e9b7 f288ae6 9e6e9b7 f288ae6 9e6e9b7 f288ae6 9e6e9b7 f288ae6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | {
"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
}
}
}
|