BogdanTurbal/synthetic_mri_generation_sd_xl

<!-- This model card has been generated automatically according to the information the training script had access to. You should probably proofread and complete it, then remove this comment. -->

BRAIN with Alzheimer's MRI generation (LoRA text2image fine-tuning)

These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. Original dataset: https://www.kaggle.com/datasets/tourist55/alzheimers-dataset-4-class-of-images.

With this model, you can generate an MRI of the brain of 4 classes: 1) NonDemented 2) VeryMildDemented 3) MildDemented 4) ModerateDemeneted

For generation use the following corresponding prompts (or any other modification)

1) MRI of NonDemented brain 2) MRI of VeryMildDemented brain 3) MRI of MildDemented brain 4) MRI of ModerateDemeneted brain

[image] [image] [image] [image] [image] [image]

LoRA for the text encoder was enabled: False.

Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.

Intended uses & limitations

How to use

# TODO: add an example code snippet for running this diffusion pipeline

Limitations and bias

[TODO: provide examples of latent issues and potential remediations]

Training details

[TODO: describe the data used to train the model]