Likalto4/Breast_unconditional_64

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