/Pokedex

Python code to scrape images for training

Primary LanguagePython

Description:

Use Bing Image search API to scrape down a dataset for training

Setup:

Get an API key here and put it in a file called apiKey.txt

Downloading dataset:

python search_bing_api.py --query "charmander" --output dataset/charmander

Training:

python train.py --dataset dataset --model pokedex.model --labelbin lb.pickle

Classify Test

python classify.py --model pokedex.model --labelbin lb.pickle --image charmander_counter.png

CoreML converter

python coremlconverter.py --model pokedex.model --labelbin lb.pickle

References:

Scraping Dataset for Training Training a Keras Model Running Keras models on iOS with CoreMLs