Skip to content

Add support for data augmentation via adapter layer to monai image transformation suite#25

Merged
wli51 merged 9 commits intoWayScience:mainfrom
wli51:dev-data-aug
Apr 23, 2026
Merged

Add support for data augmentation via adapter layer to monai image transformation suite#25
wli51 merged 9 commits intoWayScience:mainfrom
wli51:dev-data-aug

Conversation

@wli51
Copy link
Copy Markdown
Collaborator

@wli51 wli51 commented Apr 22, 2026

Adds

  1. src/virtual_stain_flow/datasets/base_wrapper_dataset.py general abstraction for lighter-weight dataset wrappers that fulfill the basic torch dataset functions without needing to fully support the package BaseImageDataset implementation which can be overwhelming.
  2. src/virtual_stain_flow/datasets/monai_aug_adapter_dataset.py realization of wrapper dataset for monai image transformation adaptation purposes
  3. examples/4.data_augmentation_example.ipynb example applying some monai transformations to dataset
  4. Other minor modifications to the visualization modules to extend support to wrapper datasets

@review-notebook-app
Copy link
Copy Markdown

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@wli51 wli51 requested a review from MattsonCam April 23, 2026 16:42
Copy link
Copy Markdown
Member

@MattsonCam MattsonCam left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good job, @wli51 ! Added a few comments, monai looks like a good tool to use here

Comment thread src/virtual_stain_flow/evaluation/evaluation_utils.py Outdated
Comment thread examples/nbconverted/4.data_augmentation_example.py
Comment thread examples/nbconverted/4.data_augmentation_example.py
Comment thread src/virtual_stain_flow/evaluation/visualization.py Outdated
Comment thread src/virtual_stain_flow/datasets/monai_aug_adapter_dataset.py Outdated
Comment thread examples/nbconverted/4.data_augmentation_example.py Outdated
Comment thread src/virtual_stain_flow/datasets/monai_aug_adapter_dataset.py
Comment thread examples/nbconverted/4.data_augmentation_example.py Outdated
Comment thread examples/nbconverted/4.data_augmentation_example.py
@wli51
Copy link
Copy Markdown
Collaborator Author

wli51 commented Apr 23, 2026

@MattsonCam thanks for reviewing, addressed most of your comments, made a issue for one that is better for a separate enhancement PR for the future. Merging now!

@wli51 wli51 merged commit c81c4e2 into WayScience:main Apr 23, 2026
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants