add RetinaUNet model into Pyhealth, expand usage with sample notebook.#979

Open

iamkishann wants to merge 48 commits intosunlabuiuc:masterfrom

iamkishann:feature/retina_unet

Commits on Mar 27, 2026

Add RetinaUNet segmentation model and register it in models package

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed

Commits on Apr 2, 2026

Expand retina U-Net implementation

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Refine RetinaUNet BaseModel-style API outputs

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed

Commits on Apr 3, 2026

Add Retina U-Net Sample code for medical image detection
VictorRosales19
authored

Commits on Apr 6, 2026

Complete Retina Unet model implementation
VictorRosales19
authored

Commits on Apr 9, 2026

Add core tests for RetinaUNet model

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed

Commits on Apr 10, 2026

Add Retina U-Net training pipeline
VictorRosales19
authored
Add dummy dataset generation for Retina U-Net
VictorRosales19
authored
Add LIDC dataset loader for PyHealth integration
VictorRosales19
authored
Add test script for Retina U-Net training pipeline
VictorRosales19
authored
Refactor class loss computation and segmentation loss
VictorRosales19
authored
Refactor retina_unet_sample.py → retina_unet.py with BaseModel wrapper

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Remove retina_unet_sample.py (migrated to retina_unet.py)

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Refactor: Rename RetinaUNetCore to RetinaUNetLayer, add comprehensive docs, remove deprecated alias and print statement

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Fix: Correct BaseModel.__init__() call - remove invalid feature_keys and label_keys parameters

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
docs: Add RetinaUNet API documentation

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Export RetinaUNetLayer for direct access alongside RetinaUNet

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Improve RetinaUNet coverage and fix 3D FPN path

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed

Commits on Apr 11, 2026

Update RetinaUNet 3D path and tests

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed

Commits on Apr 12, 2026

Integrate RetinaUNet training losses and notebook examples

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Replace shared RetinaUNet notebooks with training notebook

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Update training_retina_unet Colab sync and diagnostics

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Merge branch 'sunlabuiuc:master' into feature/retina_unet
VictorRosales19
authored

Commits on Apr 14, 2026

Add RetinaUNet segmentation model and register it in models package

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Expand retina U-Net implementation

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Refine RetinaUNet BaseModel-style API outputs

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Add Retina U-Net Sample code for medical image detection

VictorRosales19
authored and
Kishan Sarvaiya
committed
Complete Retina Unet model implementation

VictorRosales19
authored and
Kishan Sarvaiya
committed
Add core tests for RetinaUNet model

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Add Retina U-Net training pipeline

VictorRosales19
authored and
Kishan Sarvaiya
committed
Add dummy dataset generation for Retina U-Net

VictorRosales19
authored and
Kishan Sarvaiya
committed
Add LIDC dataset loader for PyHealth integration

VictorRosales19
authored and
Kishan Sarvaiya
committed
Add test script for Retina U-Net training pipeline

VictorRosales19
authored and
Kishan Sarvaiya
committed
Refactor class loss computation and segmentation loss

VictorRosales19
authored and
Kishan Sarvaiya
committed
Refactor retina_unet_sample.py → retina_unet.py with BaseModel wrapper

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Remove retina_unet_sample.py (migrated to retina_unet.py)

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Refactor: Rename RetinaUNetCore to RetinaUNetLayer, add comprehensive docs, remove deprecated alias and print statement

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Fix: Correct BaseModel.__init__() call - remove invalid feature_keys and label_keys parameters

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
docs: Add RetinaUNet API documentation

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Export RetinaUNetLayer for direct access alongside RetinaUNet

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Improve RetinaUNet coverage and fix 3D FPN path

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Update RetinaUNet 3D path and tests

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Integrate RetinaUNet training losses and notebook examples

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Replace shared RetinaUNet notebooks with training notebook

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
Update training_retina_unet Colab sync and diagnostics

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed
update retina_unet.py

Kishan Sarvaiya
authored and
Kishan Sarvaiya
committed

Commits on Apr 16, 2026