DoB24 commited on
Commit
e6249ae
·
verified ·
1 Parent(s): 33be17a

Add Quick Inference, machine-readable artefacts, and compute disclosure

Browse files
Files changed (1) hide show
  1. README.md +43 -0
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** |