/PokeRecognition-API

Pokémon image classfication using Tensorflow and FastAPI

Primary LanguagePythonApache License 2.0Apache-2.0

PokeRécognition-API

A simple Pokémon image classifier made using Keras and FastAPI.

How to Setup

  1. Install python and the required pypi packages.
  2. Scrape your Pokémon images in two folders named dataset/training and dataset/validation, sort the subfolders labeled by dex numbers.
  3. Use the trainer.py file to create a keras model.
  4. Use the main.py file to test your keras model.

Additional themes

Additional themes and templates for the website can be found here