| { | |
| "model_name": "Heartformer", | |
| "architecture": "RF-DETR Nano", | |
| "task": "object-detection", | |
| "num_classes": 8, | |
| "class_names": [ | |
| "heart-anatomy-images", | |
| "heart_cadaver", | |
| "heart_cell", | |
| "heart_ct_scan", | |
| "heart_drawing", | |
| "heart_textbook", | |
| "heart_wall", | |
| "heart_xray" | |
| ], | |
| "input_size": 640, | |
| "backbone": "DINOv2", | |
| "patch_size": 16, | |
| "num_queries": 300, | |
| "parameters": "30.5M", | |
| "training": { | |
| "dataset": "Heart Anatomy Types v2 (Roboflow)", | |
| "num_images": 621, | |
| "train_split": 497, | |
| "valid_split": 62, | |
| "test_split": 62, | |
| "epochs": 30, | |
| "best_epoch": 4, | |
| "batch_size": 4, | |
| "effective_batch_size": 16, | |
| "learning_rate": 1e-4, | |
| "optimizer": "AdamW", | |
| "lr_scheduler": "cosine_annealing_warmup", | |
| "early_stopping_patience": 8, | |
| "training_time_hours": 1.83, | |
| "hardware": "Apple M3 MacBook Pro (MPS)" | |
| }, | |
| "performance": { | |
| "map_50": 0.977, | |
| "map_50_95": 0.791, | |
| "precision_at_0.5": 0.968, | |
| "recall_at_0.5": 0.989, | |
| "f1_score_at_0.5": 0.978 | |
| }, | |
| "inference": { | |
| "default_threshold": 0.3, | |
| "recommended_threshold": 0.5, | |
| "max_detections_per_image": 300, | |
| "average_inference_time_seconds": 1.8 | |
| }, | |
| "license": "apache-2.0", | |
| "framework": "PyTorch", | |
| "version": "1.0.0", | |
| "created": "2024-10-19" | |
| } | |