Auto-image-classifier

Prototype demo👇

prototype deployment is here

what is Auto-image-classifier?

Actually, the concept is quite simple. First you need to define the categories that you want to make a classification in srape_train/fish.json. Then the Web Scraper will get photos of these categories from Bing. Then train a model for classification.

Setup

• Scrape & trian model

# first define the classification subjects in `/scrape_train/fish.json`
# then scrape web images of the subjects
cd scrape_train/
python fetch.py --limit 10  # the number of images for each class
# train model
python train_main.py 

• Deployment

cd flask_fish_reco/
python myapp.py

File structure

scrape_train/: scraping photos and training.

flask_fish_reco/: is the web deployment with Flask.

neural_vue/: user interfaces using Vue.

Prototype looks like 👉