← Back to Projects

Cat Diffusion Model

2025

Self-trained cat generation model and web app.

Technologies Used:

PythonPytorchJavaScript
Cat Diffusion Model

This is a project from one of my university courses about computer vision and deep learning. The task was to train a diffusion model from scratch for generating cats. Many different hyperparameters and model configurations needed to be tuned and benchmarked to finally receive a good image.

Final Images

Here are some example images. They differ significantly in their quality. Array of example Images of generated Cats

Demonstration

After a good model was trained, an additional web app was built together with a simple Python API to directly generate cat images on the web. Here is a short demo: