.gitignore
CHANGELOG.md
LICENSE
README.md
pyproject.toml
.github/workflows/ci.yml
.github/workflows/release.yml
.github/workflows/testpypi.yml
brainmint/__init__.py
brainmint/_version.py
brainmint/py.typed
brainmint.egg-info/PKG-INFO
brainmint.egg-info/SOURCES.txt
brainmint.egg-info/dependency_links.txt
brainmint.egg-info/requires.txt
brainmint.egg-info/top_level.txt
brainmint/callbacks/log_images.py
brainmint/callbacks/modality_completion_schedule.py
brainmint/data/__init__.py
brainmint/data/brainscape.py
brainmint/data/brainscape_paired.py
brainmint/data/utils.py
brainmint/data/transforms/__init__.py
brainmint/data/transforms/conditioning.py
brainmint/data/transforms/copy_paths.py
brainmint/data/transforms/demographics.py
brainmint/data/transforms/intensity.py
brainmint/data/transforms/modality_choice.py
brainmint/data/transforms/mri_vae.py
brainmint/data/transforms/segmentation.py
brainmint/data/transforms/stream_mapping.py
brainmint/data/transforms/synthetic_mask.py
brainmint/external/__init__.py
brainmint/external/registry.py
brainmint/external/repo_manager.py
brainmint/external/sys_path.py
brainmint/inference/__init__.py
brainmint/inference/dynamic_inference.py
brainmint/inference/controlnet/__init__.py
brainmint/inference/controlnet/controlnet_inference.py
brainmint/inference/core/__init__.py
brainmint/inference/core/context.py
brainmint/inference/core/interfaces.py
brainmint/inference/core/runner.py
brainmint/inference/core/scheduler.py
brainmint/inference/diffusion/__init__.py
brainmint/inference/diffusion/conditioning/__init__.py
brainmint/inference/diffusion/conditioning/base.py
brainmint/inference/diffusion/conditioning/common/__init__.py
brainmint/inference/diffusion/conditioning/common/class_labels.py
brainmint/inference/diffusion/conditioning/common/constant.py
brainmint/inference/diffusion/conditioning/common/demographics.py
brainmint/inference/diffusion/conditioning/common/random_vec.py
brainmint/inference/diffusion/conditioning/common/vector_ops.py
brainmint/inference/diffusion/conditioning/ldm/__init__.py
brainmint/inference/diffusion/conditioning/ldm/fixed_4vec.py
brainmint/inference/diffusion/latent/__init__.py
brainmint/inference/diffusion/latent/base.py
brainmint/inference/diffusion/latent/common/__init__.py
brainmint/inference/diffusion/latent/common/fixed_shape.py
brainmint/inference/diffusion/latent/common/from_autoencoder_encode.py
brainmint/inference/diffusion/latent/common/from_batch.py
brainmint/inference/diffusion/latent/common/from_image_shape.py
brainmint/inference/diffusion/pipelines/__init__.py
brainmint/inference/diffusion/pipelines/base.py
brainmint/inference/diffusion/pipelines/generation/__init__.py
brainmint/inference/diffusion/pipelines/generation/ldm/__init__.py
brainmint/inference/diffusion/pipelines/generation/ldm/ukb_ddim.py
brainmint/inference/diffusion/samplers/__init__.py
brainmint/inference/diffusion/samplers/base.py
brainmint/inference/diffusion/samplers/common/cond_unet.py
brainmint/inference/diffusion/samplers/common/monai_unet.py
brainmint/inference/diffusion/samplers/ldm/__init__.py
brainmint/inference/diffusion/samplers/ldm/ukb_ddim.py
brainmint/inference/generation/__init__.py
brainmint/inference/generation/batch_builders/__init__.py
brainmint/inference/generation/batch_builders/med_ddpm.py
brainmint/inference/generation/pipelines/__init__.py
brainmint/inference/generation/pipelines/common.py
brainmint/inference/generation/pipelines/external.py
brainmint/inference/generation/pipelines/hagan.py
brainmint/inference/generation/pipelines/maisi.py
brainmint/inference/generation/pipelines/med_ddpm.py
brainmint/inference/generation/pipelines/wdm3d.py
brainmint/inference/io/__init__.py
brainmint/inference/io/base.py
brainmint/inference/io/dataset_writers.py
brainmint/inference/io/readers.py
brainmint/inference/io/writers.py
brainmint/inference/postprocess/__init__.py
brainmint/inference/postprocess/base.py
brainmint/inference/postprocess/brats_pipeline.py
brainmint/inference/postprocess/reorient.py
brainmint/inference/translation/__init__.py
brainmint/inference/translation/generators/__init__.py
brainmint/inference/translation/generators/aldm.py
brainmint/inference/translation/generators/cwdm.py
brainmint/inference/translation/pipelines/__init__.py
brainmint/integrations/__init__.py
brainmint/integrations/aldm/__init__.py
brainmint/integrations/aldm/ldm.py
brainmint/integrations/aldm/repo.py
brainmint/integrations/aldm/vqgan.py
brainmint/integrations/brainsynth/__init__.py
brainmint/integrations/brainsynth/inferer.py
brainmint/integrations/brainsynth/vendor_vqvae.py
brainmint/integrations/brasyn/__init__.py
brainmint/integrations/brasyn/io.py
brainmint/integrations/brasyn/missing_mri.py
brainmint/integrations/brasyn/modalities.py
brainmint/integrations/brasyn/runtime.py
brainmint/integrations/cwdm/__init__.py
brainmint/integrations/cwdm/repo.py
brainmint/integrations/cwdm/translator.py
brainmint/integrations/hagan/__init__.py
brainmint/integrations/hagan/generator.py
brainmint/integrations/hagan/repo.py
brainmint/integrations/maisi/__init__.py
brainmint/integrations/maisi/autoencoder.py
brainmint/integrations/maisi/generator.py
brainmint/integrations/maisi/repo.py
brainmint/integrations/med_ddpm/__init__.py
brainmint/integrations/med_ddpm/generator.py
brainmint/integrations/med_ddpm/repo.py
brainmint/integrations/wdm3d/__init__.py
brainmint/integrations/wdm3d/generator.py
brainmint/integrations/wdm3d/repo.py
brainmint/lightning/__init__.py
brainmint/lightning/controlnet_module.py
brainmint/lightning/diffusion_inference_module.py
brainmint/lightning/diffusion_module.py
brainmint/lightning/export_latents_module.py
brainmint/lightning/generic_inference_module.py
brainmint/lightning/vae_module.py
brainmint/losses/__init__.py
brainmint/losses/mask_recon_loss.py
brainmint/losses/utils.py
brainmint/losses/vae_loss_manager.py
brainmint/metrics/__init__.py
brainmint/metrics/diffusion_metrics.py
brainmint/metrics/reconstruction.py
brainmint/models/__init__.py
brainmint/models/blocks/__init__.py
brainmint/models/blocks/haar_dwt.py
brainmint/models/blocks/haar_wavelet_fusion.py
brainmint/models/compression/__init__.py
brainmint/models/compression/aldm_vqgan.py
brainmint/models/compression/brainsynth_vqvae.py
brainmint/models/compression/dwt.py
brainmint/models/compression/ldm_vae.py
brainmint/models/compression/maisi_vae_gan.py
brainmint/models/compression/wavelet_fusion.py
brainmint/models/compression/wavelet_vae.py
brainmint/models/conditioning/demographics_encoder.py
brainmint/models/controlnet/__init__.py
brainmint/models/controlnet/controlnet.py
brainmint/models/controlnet/controlnet_monai.py
brainmint/models/generation/__init__.py
brainmint/models/generation/diffusion_unet.py
brainmint/models/generation/hagan.py
brainmint/models/generation/maisi.py
brainmint/models/generation/med_ddpm.py
brainmint/models/generation/wdm3d.py
brainmint/models/schedulers/__init__.py
brainmint/models/schedulers/rflow_scheduler.py
brainmint/models/translation/__init__.py
brainmint/models/translation/aldm.py
brainmint/models/translation/brasyn.py
brainmint/models/translation/cwdm.py
brainmint/models/translation/utils.py
brainmint/utils/__init__.py
brainmint/utils/batch.py
brainmint/utils/ema.py
brainmint/utils/gpumem_utils.py
brainmint/utils/schedules.py
brainmint/utils/spatial.py
brainmint/utils/state_dict_loader.py
brainmint/visualization/__init__.py
brainmint/visualization/slices.py
tests/README.md
tests/conftest.py
tests/data/test_brainscape.py
tests/data/test_brainscape_latent_variants.py
tests/data/test_brainscape_latents.py
tests/data/test_brainscape_paired.py
tests/data/test_demographics_transform.py
tests/data/test_export_brainscape_latents.py
tests/data/test_modality_choice_transform.py
tests/data/test_synthetic_mask_transform.py
tests/data/test_translation_configs.py
tests/external/test_repo_manager.py
tests/losses/test_vae_losses.py
tests/metrics/test_report_metrics_utils.py
tests/models/test_demographics_encoder.py
tests/models/test_maisi_controlnet.py
tests/models/test_maisi_diffusion.py
tests/models/test_maisi_vae.py
tests/models/test_wavelet_fusion_vae.py
tests/package/test_import.py
tests/package/test_import_surfaces.py
tests/trainer/test_controlnet.py
tests/trainer/test_diffusion.py
tests/trainer/test_vae.py
tests/utils/test_schedules.py
tests/utils/test_state_dict_loader.py