Project-MONAI / Project-MONAI/model-zoo
Use CacheDataset for generative bundles.
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 338
- Forks
- 106
- PR merge metrics
- No merged PRs in 30d
Description
Is your feature request related to a problem? Please describe.
For the diffusion bundle, monai.apps.DecathlonDataset is used by the generative bundles.
It provides some convenience (auto-downloading datasets), but it makes overwriting the input datalist more difficult. Can we replace it with CacheDataset like the other bundles, which have more flexibility?
Describe the solution you'd like
Use CacheDataset for generative bundles.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the dataset definitions in models/brats_mri_generative_diffusion/configs/train_autoencoder.json and models/brats_mri_axial_slices_generative_diffusion/configs/train_diffusion.json, then compare them with the other bundles that use CacheDataset. Replace the generative bundle dataset configuration so the input datalist can be overridden, and confirm both bundle configs still load correctly.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 2/5
- Estimated time
- 1-3 hours
- Activity status
- Stale
- Clarity
- Clearly specified
- Newbie friendliness
- 55/100