Skip to main content

Installer to help to use the prototypes from MONAI generative models in other projects.

Project description

project-monai

MONAI Generative Models

Prototyping repository for generative models to be integrated into MONAI core, MONAI tutorials, and MONAI model zoo.

Features

  • Network architectures: Diffusion Model, Autoencoder-KL, VQ-VAE, Autoregressive transformers, (Multi-scale) Patch-GAN discriminator.
  • Diffusion Model Noise Schedulers: DDPM, DDIM, and PNDM.
  • Losses: Adversarial losses, Spectral losses, and Perceptual losses (for 2D and 3D data using LPIPS, RadImageNet, and 3DMedicalNet pre-trained models).
  • Metrics: Multi-Scale Structural Similarity Index Measure (MS-SSIM) and Fréchet inception distance (FID).
  • Diffusion Models, Latent Diffusion Models, and VQ-VAE + Transformer Inferers classes (compatible with MONAI style) containing methods to train, sample synthetic images, and obtain the likelihood of inputted data.
  • MONAI-compatible trainer engine (based on Ignite) to train models with reconstruction and adversarial components.
  • Tutorials including:
    • How to train VQ-VAEs, VQ-GANs, VQ-VAE + Transformers, AutoencoderKLs, Diffusion Models, and Latent Diffusion Models on 2D and 3D data.
    • Train diffusion model to perform conditional image generation with classifier-free guidance.
    • Comparison of different diffusion model schedulers.
    • Diffusion models with different parameterizations (e.g., v-prediction and epsilon parameterization).
    • Anomaly Detection using VQ-VAE + Transformers and Diffusion Models.
    • Inpainting with diffusion model (using Repaint method)
    • Super-resolution with Latent Diffusion Models (using Noise Conditioning Augmentation)

Roadmap

Our short-term goals are available in the Milestones section of the repository.

In the longer term, we aim to integrate the generative models into the MONAI core repository (supporting tasks such as, image synthesis, anomaly detection, MRI reconstruction, domain transfer)

Installation

To install the current release of MONAI Generative Models, you can run:

pip install monai-generative

To install the current main branch of the repository, run:

pip install git+https://github.com/Project-MONAI/GenerativeModels.git

Requires Python >= 3.8.

Contributing

For guidance on making a contribution to MONAI, see the contributing guidelines.

Community

Join the conversation on Twitter @ProjectMONAI or join our Slack channel.

Citation

If you use MONAI Generative in your research, please cite us! The citation can be exported from the paper.

Links

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

monai-generative-0.2.3.tar.gz (114.6 kB view details)

Uploaded Source

Built Distribution

monai_generative-0.2.3-py3-none-any.whl (176.2 kB view details)

Uploaded Python 3

File details

Details for the file monai-generative-0.2.3.tar.gz.

File metadata

  • Download URL: monai-generative-0.2.3.tar.gz
  • Upload date:
  • Size: 114.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.11.6

File hashes

Hashes for monai-generative-0.2.3.tar.gz
Algorithm Hash digest
SHA256 0f54719c25fbc498a068004876a04c8bef771d968b013c0af378a5a4fc9a47d7
MD5 8d892d073f8f9182d0bc00297fc25d8c
BLAKE2b-256 4aadfcc3cb208da7834c57f7e40a746e07dd835c7e6580568e9c025eb36d8780

See more details on using hashes here.

File details

Details for the file monai_generative-0.2.3-py3-none-any.whl.

File metadata

File hashes

Hashes for monai_generative-0.2.3-py3-none-any.whl
Algorithm Hash digest
SHA256 44dee6264e72f2e7577ad41104db48fae2ca701a3c35e2c74c7d14133b5a16a1
MD5 d0e33d3ca42b7c5b6bcefec3234ff352
BLAKE2b-256 3937d1f9e8efbd68b69edf791318ce9ced327ced4850896afe24f687a582204f

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page