Add Quick Inference, machine-readable artefacts, and compute disclosure
Browse files
README.md
CHANGED
|
@@ -363,3 +363,46 @@ Per-class Grad-CAM saliency maps (one PNG per CNN-class model, 10 classes per fi
|
|
| 363 |
| VGG19 | 79.31 ± 1.89% |
|
| 364 |
| ResNet101 | 79.27 ± 1.07% |
|
| 365 |
| CLIP Transformer | 63.92 ± 1.79% |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 363 |
| VGG19 | 79.31 ± 1.89% |
|
| 364 |
| ResNet101 | 79.27 ± 1.07% |
|
| 365 |
| CLIP Transformer | 63.92 ± 1.79% |
|
| 366 |
+
|
| 367 |
+
|
| 368 |
+
## Quick Inference
|
| 369 |
+
|
| 370 |
+
```bash
|
| 371 |
+
pip install torch torchvision timm open_clip_torch huggingface_hub pillow
|
| 372 |
+
python code/inference_example.py path/to/fundus.jpg --model densenet121
|
| 373 |
+
```
|
| 374 |
+
|
| 375 |
+
The script downloads the requested checkpoint from this repo via the HF Hub, applies the
|
| 376 |
+
same 224x224 + ImageNet-normalisation preprocessing used during evaluation, and prints
|
| 377 |
+
the top-3 predicted classes with probabilities.
|
| 378 |
+
|
| 379 |
+
## Dataset Figures
|
| 380 |
+
|
| 381 |
+
- `dataset/class_distribution.png` + `.csv` — per-class counts, original vs augmented
|
| 382 |
+
- `dataset/sample_grid.png` — one representative image per class
|
| 383 |
+
|
| 384 |
+
## Machine-Readable Artefacts
|
| 385 |
+
|
| 386 |
+
- `code/hparams.json` — full hyperparameter table (optimizer, schedule, augmentations, TTA, ensemble, per-model LR, GPU hours)
|
| 387 |
+
- `analysis/per_class_metrics.csv` — precision/recall/F1/support per model per class
|
| 388 |
+
- `analysis/confusion_matrices_csv/cm_<model>.csv` — numeric confusion matrices
|
| 389 |
+
- `CITATION.cff` — Citation File Format
|
| 390 |
+
|
| 391 |
+
## Compute Disclosure
|
| 392 |
+
|
| 393 |
+
All 9 models + ensemble were trained on a single **NVIDIA Tesla T4 (16 GB)** under
|
| 394 |
+
PyTorch 2.11.0+cu128. Approximate wall-clock GPU-hours:
|
| 395 |
+
|
| 396 |
+
| Model | GPU-hours |
|
| 397 |
+
|---|---:|
|
| 398 |
+
| VGG19 | 2.5 |
|
| 399 |
+
| ResNet50 | 2.3 |
|
| 400 |
+
| ResNet101 | 3.1 |
|
| 401 |
+
| DenseNet121 | 2.8 |
|
| 402 |
+
| InceptionV3 | 2.2 |
|
| 403 |
+
| Swin-B | 4.5 |
|
| 404 |
+
| CLIP-ViT-B/16 | 3.8 |
|
| 405 |
+
| DINOv2-L | 11.2 |
|
| 406 |
+
| RETFound | 9.4 |
|
| 407 |
+
| Ensemble + stats | 0.3 |
|
| 408 |
+
| **Total** | **~42 GPU-hours** |
|