Skip to content

Implementation of Hot Dog or not app from Silicon Valley (CNN to identify if the given picture is a hot dog or not)

Notifications You must be signed in to change notification settings

VPanjeta/hotdog-or-not-hotdog

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

hotdog-or-not-hotdog

Implementation of Hot Dog or not app from Silicon Valley (CNN to identify if the given picture is a hot dog or not)

I used Google's Inception (2015) model. Inception is a deep convolutional neural network built for classifying real world images of thousand category.
Retraining done by replacing last layer of Inception model.
Training has been done using around 300 images so the accuracy for certain images might be low.

Getting started

  1. Clone and run bash on this repository. (Ensure that you have TensorFlow installed.)
  2. Then run python label_dog.py test/bottle.jpg bottle
  3. Wait for the model's prediction.
  4. Here's the result you should expect if things work correctly:
not hot dog          : 0.93821
hot dog              : 0.06179
  1. For hot dog run python label_dog.py test/hotdog.jpg hotdog
  2. The result comes out as
hot dog              : 0.99862
not hot dog          : 0.00138