metadata
license: mit
tags:
- pytorch
- diffusers
- unconditional-image-generation
- diffusion-models-class
This model is a diffusion model for unconditional image generation of mammograms of size 64x64 pixels. The model was trained with 1000 images using the DDPM architecture. The model was trained for 50 epochs with a batch size of 64, using around 11 GB of GPU memory.
Usage
from diffusers import DDPMPipeline
pipeline = DDPMPipeline.from_pretrained({hub_model_id})
image = pipeline().images[0]
image