Logo
PokeGen-AI - Génération de Pokémon par IA
2026
Projet personnel
Featured

PokeGen-AI

Générateur d'images de Pokémon par IA générative comparant plusieurs modèles Stable Diffusion fine-tunés avec LoRA.

Technologies

PythonPyTorchHugging Face DiffusersLoRAStable DiffusionWeights & BiasesGradio

Aperçu du projet

PokeGen-AI compare plusieurs modèles Stable Diffusion (1.5, 2.1, XL, 3.5) fine-tunés avec LoRA pour la génération d'images de Pokémon. Le projet inclut une implémentation GAN comme baseline de comparaison, un pipeline de création de dataset avec fusion d'arrière-plans, une interface Gradio pour la génération interactive et un suivi des expériences via Weights & Biases.

Mes actions & réalisations

1

Fine-tuning LoRA sur Stable Diffusion 1.5, 2.1, XL et 3.5

2

Création d'un pipeline de dataset avec fusion d'arrière-plans

3

Implémentation d'un GAN comme baseline de comparaison

4

Développement d'une interface Gradio pour la génération interactive

5

Suivi et analyse des expériences via Weights & Biases

6

Comparaison quantitative et qualitative des modèles générés

Compétences développées

Fine-tuning de modèles génératifs (LoRA / PEFT)Maîtrise de l'écosystème Hugging Face DiffusersPréparation et augmentation de datasetsMonitoring d'expériences ML (W&B)Évaluation et comparaison de modèles génératifs