search

Project-MONAI / model-zoo

MONAI Model Zoo that hosts models in the MONAI Bundle format.
Apache License 2.0
173 stars 63 forks source link

Use CacheDataset for generative bundles. #544

Open mingxin-zheng opened 6 months ago

mingxin-zheng commented 6 months ago

Is your feature request related to a problem? Please describe.

For the diffusion bundle, monai.apps.DecathlonDataset is used by the generative bundles.

https://github.com/Project-MONAI/model-zoo/blob/f9334c14dbd779898b4ec04ac2f72554b486a668/models/brats_mri_generative_diffusion/configs/train_autoencoder.json#L143

https://github.com/Project-MONAI/model-zoo/blob/f9334c14dbd779898b4ec04ac2f72554b486a668/models/brats_mri_axial_slices_generative_diffusion/configs/train_diffusion.json#L105

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.