Skip to content

thaynnara007/pokedex

Repository files navigation

Pokedex

Pokedex it's a artificial intelligence that has as goal recognizing some pokemons images.

This is a project that was developed for Inteligência Artificial course from UFCG


Usage

  • The model produced can classify seven pokemons, which are: pikachu, charmander, squirtle, bulbasaur, cyndaquil, totodile and chikorita.
  • The searchimages package has some scripts that can be used for collecting images for the dataset.
  • Since the trained model is too big for a commit , it will be find on google drive.
  • The model was trained with a dataset of about 8490 images.
  • Trained model

Set up

Creating the virtual environment

$ python -m venv pokevenv

Install dependencies

$ pip install -r requirements.txt

Activating virtual envrionment

$ . pokevenv/bin/activate

Using

To classify images

$ python3 classifier.py --model path_to_pokedex.model --labelbin lb.pickle --image examples/pikachu1.jpeg

To train a new model

$ python3 train.py --dataset path_to_dataset --model path_to_pokedex.model --labelbin lb.pickle

For more details about the arguments

$ python3 classifier.py --help
$ python3 train.py --help

Results

Graphics

  • This is the graphic of the first model that was trained. first model

  • This is the graphic of the last model that was trained; this one had a bigger dataset. last model


Classification

  • Here there are some examples of successful images classification
  • There are more examples of classification in examples package
  • Pikachu pikachu
  • Charmander charmander
  • Bulbasaur bulbasaur
  • Squirtle squirtle
  • Cyndaquil cyndaquil
  • Chikorita chikorita
  • Totodile totodile

Built with

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages