Downloads · 30 days
16
2% of all-time downloads
Likalto4/Breast_unconditional_64
Breast_unconditional_64 is a unconditional image generation model from Likalto4. Use it for the unconditional image generation task on the model card, and read the license before you ship it in a product. It is set up for diffusers. The card lists the license as mit.
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 bat…
Downloads · 30 days
16
2% of all-time downloads
All-time downloads
748
Public
Repo size
910 MB
Likes
0
Public
Click a slice to open those files.
.bin455 MB · 100%
From the Hugging Face model README
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.
from diffusers import DDPMPipeline
pipeline = DDPMPipeline.from_pretrained({hub_model_id})
image = pipeline().images[0]
image