Dieses Projekt wurde bei le Wagon in zwei Wochen während des Data Science Bootcamps durchgeführt. Mit diesem Projekt können Sie dank eines CNN-Modells die 151 ersten Taschenmonster (AKA-Pokémon) nach Typen und Namen klassifizieren. Aber das ist es nicht! Dank eines GAN-Modells können Sie auch ein neues generieren.
Um die App zu testen, besuchen Sie diese Website: https://pokemon-generator-1672.streamlit.app/
Zuerst klonen wir das Repository:
git clone https://github.com/Just-PH/lewagon-pokedex-gan.git
Führen Sie dann die Installation aus:
cd backend
make start
Noch in /backend So testen Sie die beiden Vorhersagefunktionen für alle Bilder:
make run_test
Wenn Sie nur die Typen testen möchten:
make run_test_15
Wenn Sie nur die Namen testen möchten:
make run_test_150
Um sie vorherzusagen:
make run_pred
Wenn Sie nur für die Typen vorhersagen möchten:
make run_pred_15
Wenn Sie nur die Namen vorhersagen möchten:
make run_pred_150
Um diese Art von Bildern zu erzeugen:
Sie können diesen Befehl verwenden:
make run_generate
Das Bild befindet sich im Repository mit dem Namen „output_gan“.
So führen Sie die API lokal aus:
make run_api_local