Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+dataset_preview.png filter=lfs diff=lfs merge=lfs -text
+seg_predictions.png filter=lfs diff=lfs merge=lfs -text
+seg_training_curves.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,23 @@
+---
+license: mit
+tags:
+- medical-imaging
+- retinal-segmentation
+- retinopathy-of-prematurity
+- pytorch
+- unet
+- segmentation
+---
+
+# ROP Retinal Segmentation — U-Net (ResNet50)
+
+Multi-class segmentation of retinal structures in ROP screening images.  
+Dataset: HVDROPDB | Classes: Background · Vessel · OpticDisc · Ridge
+
+## Test Set Metrics
+| Structure | IoU | Dice |
+|-----------|-----|------|
+| Vessel    | 0.1987 | 0.3280 |
+| OpticDisc | 0.4838 | 0.6389 |
+| Ridge     | 0.5546 | 0.6925 |
+| **Mean**  | **0.4124** | **0.5531** |

config.json

@@ -0,0 +1,32 @@
+{
+  "model": "UNet-ResNet50",
+  "encoder": "resnet50",
+  "classes": [
+    "Background",
+    "Vessel",
+    "OpticDisc",
+    "Ridge"
+  ],
+  "img_size": 256,
+  "epochs": 40,
+  "batch_size": 8,
+  "lr": 0.0001,
+  "test_metrics": {
+    "mean_iou": 0.41239543797241324,
+    "mean_dice": 0.5531446056233512,
+    "per_class": {
+      "Vessel": {
+        "iou": 0.1987320470313231,
+        "dice": 0.32802613576253253
+      },
+      "OpticDisc": {
+        "iou": 0.48382218430439633,
+        "dice": 0.6388623962799708
+      },
+      "Ridge": {
+        "iou": 0.5546320825815201,
+        "dice": 0.6925452848275503
+      }
+    }
+  }
+}

seg_metrics.csv

@@ -0,0 +1,5 @@
+Structure,IoU,Dice
+Vessel,0.1987,0.3280
+OpticDisc,0.4838,0.6389
+Ridge,0.5546,0.6925
+MEAN,0.4124,0.5531

unet_resnet50.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b4e2db2ef4cfe7bee73d39e5da89d1e4638b700050dfc9831b93f141b0a6cc80
+size 130442909