LeWagon_Pokedex

Pokedex_project_V1.672 chez Le Wagon

Ce projet a été réalisé au Wagon en deux semaines, lors du Data Science Bootcamp. Avec ce projet, vous pouvez classer par types et noms les 151 premiers monstres de poche (AKA Pokémon), grâce à un modèle CNN. Mais ce n'est pas ça ! Vous pouvez également en générer un nouveau grâce à un modèle GAN.

Pour tester l'application, rendez-vous sur ce site : https://pokemon-generator-1672.streamlit.app/

Installation

Commençons par cloner le dépôt :

 git clone https://github.com/Just-PH/lewagon-pokedex-gan.git

Ensuite, lancez l'installation :

 cd backend
make start

️ Ensuite vous devez modifier dans le fichier /backend/.env la variable WHO avec votre nom ️

Test

Toujours dans /backend Pour tester les deux fonctions de prédictions sur toutes les images :

 make run_test

Si vous souhaitez uniquement tester les types :

 make run_test_15

Si vous souhaitez uniquement tester les noms :

 make run_test_150

Prédictions

️ METTRE toutes les images que vous souhaitez tester dans le référentiel all_prediction_images/images ️

Pour les prédire :

 make run_pred

Si vous souhaitez uniquement prédire pour les types :

 make run_pred_15

Si vous souhaitez uniquement prédire pour les noms :

 make run_pred_150

Génération

Pour générer ce genre d'images :

Vous pouvez utiliser cette commande :

 make run_generate

L'image sera dans le référentiel nommé output_gan

API

Pour exécuter localement l'API :

 make run_api_local