Skip to content


Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?

Latest commit


Git stats


Failed to load latest commit information.
Latest commit message
Commit time

Who's That Pokémon?

Who's That Pokémon

This app answers the age-old question that all Trainers have been asking since they were 10. Give it an image and it will give its best guess as to what Pokémon was in the image.

The model is pretrained in Tensorflow and then converted to a Tensorflow.js model. This model is then downloaded into your browser and loaded via Tensorflow.js. Thus, all predictions are done solely in-browser.

Uploading An Image

There are 3 ways to upload image:

  1. Upload an image: Drag an image or click to upload.
  2. Copy an image: Copy an image to your clipboard and paste in the textbox.
  3. Draw an image: Flex your artistic muscles and try to draw an Pokémon!

Training the model

To train the model, I used the Kaggle dataset of 1st generation Pokémon images found here. Augmentation was done in preprocessing to reduce overfitting.

I first trained it on ResNet50 with an input size of 224x224. Then, I halved the dimensions which resulted in significantly faster training and evaluation times. However, ResNet's model size is too large for prediction over the browser so I swapped over to MobileNet with an input size of 128x128.


Each model is evaluated over a small test dataset of 1309 images.

Model Total Time Time per step Loss Accuracy Inference Speed
ResNet-50 112x112 10s 465ms 2.6647 0.5218 0.0425s
MobileNet 128x128 10s 483ms 1.6229 0.6348 0.0278s
MobileNet 224x224 18s 874ms 1.3993 0.6578 0.135s
ResNet-50 224x224 19s 914ms 2.1577 0.5569 0.0464s

Next Steps

The current model performs poorly on drawn images. A possible direction is to preprocess the images to remove colour to detect shape or train it on hand-drawn images of Pokémon.