Image classification using TensorFlow (Inception v3) that labels any JPEG image as "hot dog" or "not a hot dog".
Inspired by Jian Yang's "Shazaam for food" from Silicon Valley: Season 4 Episode 4.
- Clone repository.
git clone https://github.com/TheDoctor0/NotAHotDog.git
- Install requirements.
python -m pip install -r requirements.txt
- Get Tensorflow Inception v3.
curl -LO https://github.com/tensorflow/hub/raw/master/examples/image_retraining/retrain.py
- Retrain model.
python retrain.py --bottleneck_dir=bottlenecks --model_dir=inception --summaries_dir=training_summaries/long --output_graph=retrained_graph.pb --output_labels=retrained_labels.txt --image_dir=images
- Use classifier on any JPEG image.
python checker.py image_path.jpg
$ python checker.py dog.jpg
not a hot dog (98.92%)