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

Draft

iamkishann wants to merge 23 commits intosunlabuiuc:masterfrom

iamkishann:feature/retina_unet

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

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

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

add RetinaUNet model into Pyhealth, expand usage with sample notebook.#979
iamkishann wants to merge 23 commits intosunlabuiuc:masterfrom
iamkishann:feature/retina_unet

Commits on Apr 14, 2026

Add RetinaUNet segmentation model and register it in models package

Expand retina U-Net implementation

Refine RetinaUNet BaseModel-style API outputs

Add Retina U-Net Sample code for medical image detection

Complete Retina Unet model implementation

Add core tests for RetinaUNet model

Add Retina U-Net training pipeline

Add dummy dataset generation for Retina U-Net

Add LIDC dataset loader for PyHealth integration

Add test script for Retina U-Net training pipeline

Refactor class loss computation and segmentation loss

Refactor retina_unet_sample.py → retina_unet.py with BaseModel wrapper

Remove retina_unet_sample.py (migrated to retina_unet.py)

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

Fix: Correct BaseModel.init() call - remove invalid feature_keys and label_keys parameters

docs: Add RetinaUNet API documentation

Export RetinaUNetLayer for direct access alongside RetinaUNet

Improve RetinaUNet coverage and fix 3D FPN path

Update RetinaUNet 3D path and tests

Integrate RetinaUNet training losses and notebook examples

Replace shared RetinaUNet notebooks with training notebook

Update training_retina_unet Colab sync and diagnostics

update retina_unet.py