Skip to content

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

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
iamkishann wants to merge 23 commits into
base: master
Choose a base branch
from
Draft

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

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
23 commits
Select commit Hold shift + click to select a range
7476c39
Add RetinaUNet segmentation model and register it in models package
Mar 27, 2026
c99d22d
Expand retina U-Net implementation
Apr 2, 2026
6d30cfa
Refine RetinaUNet BaseModel-style API outputs
Apr 2, 2026
b6f606a
Add Retina U-Net Sample code for medical image detection
VictorRosales19 Apr 3, 2026
1c48526
Complete Retina Unet model implementation
VictorRosales19 Apr 6, 2026
b7612fd
Add core tests for RetinaUNet model
Apr 9, 2026
65e9256
Add Retina U-Net training pipeline
VictorRosales19 Apr 10, 2026
a35402c
Add dummy dataset generation for Retina U-Net
VictorRosales19 Apr 10, 2026
6d3f4ea
Add LIDC dataset loader for PyHealth integration
VictorRosales19 Apr 10, 2026
cdffa5f
Add test script for Retina U-Net training pipeline
VictorRosales19 Apr 10, 2026
e16086e
Refactor class loss computation and segmentation loss
VictorRosales19 Apr 10, 2026
4d8c5af
Refactor retina_unet_sample.py → retina_unet.py with BaseModel wrapper
Apr 10, 2026
f8e3011
Remove retina_unet_sample.py (migrated to retina_unet.py)
Apr 10, 2026
2d1da8a
Refactor: Rename RetinaUNetCore to RetinaUNetLayer, add comprehensive…
Apr 10, 2026
b5edb7e
Fix: Correct BaseModel.__init__() call - remove invalid feature_keys …
Apr 10, 2026
97e46ed
docs: Add RetinaUNet API documentation
Apr 10, 2026
5341b72
Export RetinaUNetLayer for direct access alongside RetinaUNet
Apr 10, 2026
3a300ce
Improve RetinaUNet coverage and fix 3D FPN path
Apr 10, 2026
a322fae
Update RetinaUNet 3D path and tests
Apr 11, 2026
45c4859
Integrate RetinaUNet training losses and notebook examples
Apr 12, 2026
bb72adf
Replace shared RetinaUNet notebooks with training notebook
Apr 12, 2026
8637227
Update training_retina_unet Colab sync and diagnostics
Apr 12, 2026
763d4a5
update retina_unet.py
Apr 14, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -138,5 +138,12 @@ data/physionet.org/
# VSCode settings
.vscode/

# Generated files under examples/
examples/dummy_data*/
examples/checkpoints*/
examples/logs*/
examples/__pycache__/
examples/**/__pycache__/

# Model weight files (large binaries, distributed separately)
weightfiles/
1 change: 1 addition & 0 deletions docs/api/models.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -199,6 +199,7 @@ API Reference
models/pyhealth.models.TFMTokenizer
models/pyhealth.models.GAN
models/pyhealth.models.VAE
models/pyhealth.models.RetinaUNet
models/pyhealth.models.SDOH
models/pyhealth.models.VisionEmbeddingModel
models/pyhealth.models.TextEmbedding
Expand Down
14 changes: 14 additions & 0 deletions docs/api/models/pyhealth.models.RetinaUNet.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
pyhealth.models.RetinaUNet
===================================

Retina U-Net model for medical image object detection and segmentation.

.. autoclass:: pyhealth.models.RetinaUNetLayer
:members:
:undoc-members:
:show-inheritance:

.. autoclass:: pyhealth.models.RetinaUNet
:members:
:undoc-members:
:show-inheritance:
Loading