PawelLawrynowicz / pokemon-classification

Pokemon classification with resnet model

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Training and evaluation pipeline of an image classifier (Resnet50) for images of Pokemon. Code can be easily remade to preform data cleaning, augmentation, training and testing for any given dataset.

Achieved accuracy: 77%

Example of classified images can be found in the show_results.ipynb notebook.

About

Pokemon classification with resnet model


Languages

Language:Jupyter Notebook 98.2%Language:Python 1.8%