PokeGen-AI

Génération d'images de Pokémon par intelligence artificielle générative.

Description

Ce projet 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. Il inclut également une implémentation GAN comme baseline de comparaison.

Exemples de génération

Stable Diffusion 1.5	Stable Diffusion XL

Technologies

• Framework : PyTorch, Hugging Face Diffusers
• Fine-tuning : LoRA (PEFT)
• Modèles : Stable Diffusion 1.5, 2.1, XL, 3.5
• Monitoring : Weights & Biases
• Interface : Gradio

Fonctionnalités

• Fine-tuning LoRA sur plusieurs versions de Stable Diffusion
• Pipeline de création de dataset avec fusion d'arrière-plans
• Interface Gradio pour la génération interactive
• Suivi des expériences via Weights & Biases

Auteurs

Projet académique - UTBM AI54