FlavioRubensOttaviani/DiffusionModel-WelcomeToTheJungle-DDPM-v1

🐯 Welcome to the Jungle – AI Image Generator (DDPM v1)

🎯 This is a pretrained DDPM model that generates realistic 128x128 wild animal images from pure noise.

---

🎨 What You Can Use This Model For / A cosa serve questo modello

🚀 English

This pretrained DDPM model generates high-quality 128x128 images of wild animals from pure noise.
Thanks to its stable U-Net backbone and diffusion pipeline, it can be used for multiple purposes:

• 🧪 Data Augmentation: enrich your training datasets with synthetic animal images.
• 🧑‍🏫 AI Education & Demos: perfect for demonstrating generative models in lessons or workshops.
• 🖼️ Creative Projects & Art: use the generator to create textures, digital artworks, or backgrounds.
• 🧠 Research: test new sampling strategies, noise schedulers, or conditioning mechanisms.
• 🗂️ Presentation Material: generate consistent wildlife images for PowerPoint slides, mockups or publications.

---

🇮🇹 Italiano

Questo modello DDPM preaddestrato genera immagini realistiche di animali selvatici a 128x128 pixel a partire da puro rumore.
Grazie all’architettura UNet e alla pipeline di diffusione, può essere utilizzato in molti contesti:

• 🧪 Data Augmentation: arricchisci i tuoi dataset con immagini sintetiche di animali.
• 🧑‍🏫 Didattica & Dimostrazioni: ideale per spiegare i modelli generativi in corsi o workshop.
• 🖼️ Progetti Creativi & Arte Digitale: genera texture, sfondi o arte concettuale.
• 🧠 Ricerca: testa nuove tecniche di campionamento, scheduler di rumore o meccanismi condizionati.
• 📊 Presentazioni & Slide: crea immagini coerenti da usare in PowerPoint, tesi, progetti aziendali.

---

🖼️ Sample Output

🧠 Model Architecture

---

🔧 Quick Inference

Clone the repository or download the files and run:

pip install torch torchvision pillow
python inference.py


> The model will generate an image from pure noise and save it as output.png.  
> UNet and Diffusion modules are embedded in the model

### 📖 Description / Descrizione

**English:**  
Welcome to *Welcome to the Jungle – AI Image Generator (DDPM v1)*, a diffusion-based model built upon a custom UNet architecture with self-attention.
This model generates 128×128 RGB images of wild animals from random noise.  
The code has been intentionally left uncovered to provide an opportunity for learning and experimentation — feel free to study and modify it.  
Note that the generation process is inherently random, and outputs may vary between runs.  
A diffusion model works by progressively denoising a random tensor: at each time step, it predicts and subtracts the noise, moving closer to a coherent image.
The UNet architecture used here includes downsampling layers with skip connections, self-attention modules to enhance feature learning, and upsampling layers
to reconstruct the final image.

**Italiano:**  
Benvenuti in *Welcome to the Jungle – AI Image Generator (DDPM v1)*, un modello basato sulla diffusione costruito su un’architettura UNet personalizzata con moduli di
self-attention.  
Il modello genera immagini RGB 128×128 di animali selvatici partendo da rumore casuale.  
Il codice è stato volutamente lasciato visibile per offrire un'opportunità formativa a chi desidera imparare e sperimentare.  
Il processo di generazione è intrinsecamente randomico: ogni immagine può essere diversa a ogni esecuzione.  
Un modello di diffusione lavora rimuovendo progressivamente il rumore da un tensore casuale, prevedendo e sottraendo il rumore passo dopo passo, fino a ottenere
un’immagine coerente. L'architettura UNet impiegata include livelli di downsampling con connessioni skip, moduli di self-attention per migliorare l’apprendimento
delle caratteristiche e livelli di upsampling per ricostruire l'immagine finale.

---

## 🧪 Quick Inference Example

You can run the following script to generate a synthetic wildlife image using this model:

```python
import torch
from torchvision.utils import save_image
from unet import Unet, Diffusion
import os

# Setup device e immagine
# Setup device and image size

device = 'cuda' if torch.cuda.is_available() else 'cpu'
image_size = 128

# Inizializzo modello e carica pesi
model = Unet().to(device)
model.load_state_dict(torch.load("ckpt_epoch_149.pt", map_location=device))
model.eval()

# Inizializzo classe di diffusione
diffusion = Diffusion(img_size=image_size, device=device)

# Creo la cartella output se non esiste
os.makedirs("generated_images", exist_ok=True)

# Genero immagine e salva
# Generate and save image
with torch.no_grad():
    sampled_images = diffusion.sample(model, n=1)
    save_image(sampled_images, "generated_images/output.png")

print("✅ Immagine generata e salvata in 'generated_images/output.png'")

---

- Dove trovo l’immagine generata?
+ 📂 Dove viene salvata l’immagine generata?

Where to find the result?
After running the script, you will find your generated image inside the generated_images/ folder, located in the same directory as your script.

💡 Make sure the script, unet.py, and the model weights file (ckpt_epoch_149.pt) are in the same folder or adjust the import paths accordingly.

🇮🇹 Dove trovo l’immagine generata?
Una volta eseguito lo script, l’immagine viene salvata automaticamente nella cartella generated_images/, accanto al file .py con cui hai lanciato l’inferenza.

---

**🧑‍💻 Author / Autore:**  
Dr. Flavio Rubens Ottaviani

**📄 License / Licenza:**  
MIT License