Build a TensorFlow Image Classifier in 5 Min

Key Takeaways

hello world it's SJ in this episode we're going to build an image classifier using tensorflow in 30 lines of python and I don't mean a classifier that can detect handwritten digits or Iris Flowers I'm talking literally anything you want you'll be able to train this thing to classify chocolate if you want it [Music] h the possibilities are endless there's so many industries that can be disrupted by just this simple solution a Japanese cucumber farmer built a machine to detect whether each of his cucumbers was one of nine different types using this thing let's say we want to build a sage classifier if we think about this problem in the traditional programming Paradigm we want to handcraft a bunch of features maybe we could do some Edge detection to save the shape of my hair or use a color histogram to save the color of my teeth the problem with that is there's so much variance in sjas my hair is alive seriously it's never the same this is where a comp evolutional neural networks come into play they're essentially a black box that constructs features that we would otherwise have to handcraft and these abstract features they create from training are so generalized that they account for variance if we wanted to train a CNN ourselves we need a lot of computing power and a lot of time both of which we don't have I don't even have time to do my dishes sorry remits that's why we want to use a pre-trained CNN model called Inception Inception was trained by Google on 100K images with a thousand categories our use case in this video will be classifying Darth Vader pictures but Inception wasn't trained on Vader so we're going to perform a process called transfer learning that means applying the learnings from a previous training session to a new training session if we look at the Inception model we can see that when we feed in an image as an input at each layer it will perform a series of operations on that data until it outputs a label and classification percentage each layer is a different set of abstractions in the first layers it's basically taught itself Edge detection then shape detection in the middle layers and they get increasingly more abstract up until the end if we look at the last few layers these are the highest level detectors for whole objects for transfer learning we basically just want to retrain that last layer on features of Darth Vader so it can add a representation of him to its repository of knowledge so this is going to be a seven-step process and we're going to go through each step in order sound good step one is to install Docker which is a tool for creating a virtual container on your machine for running apps the benefit of Docker is you don't have to install any dependencies on your machine so we'll eventually download a Docker image that has all the necessary dependencies for tensor flow built in just download the docker toolbox go through the installation process and then you can launch your Docker container anytime easily by double clicking the docker quickart terminal cool now that we have Docker open that brings us to step two installing the tensorflow docker image by pasing in this line it'll take a few minutes then once it's installed we'll move on to step three downloading our image data set to our local machine we'll stop Docker with contr d then create a directory called TF file Star Wars in our home directory locally we want to put a folder labeled Darth Vader that contains a couple hundred Vader p Pi in here there's this dope Chrome extension I found called faton batch download image that bulk downloads all the images from your Google image search results just go to Google image search type in Darth Vader and start downloading all of those images once we've got them we'll just drag that folder into our TF file Star Wars folder that brings us to step four now that we have our images in our TF files directory we want to link them to our Docker container with this command boom all Linked UP step five is to download the training script via git just CD into the tensorflow directory then run git hole this code will allow us to retrain the Inception classifier with a newly linked Darth Vader image data set step six is the actual retraining part the bottleneck directory will be used to Cache the outputs of the lower layers on disk so they don't have to repeatedly be recalculated we'll run this example for 500 iterations the next flag asks where to store our trained model our output craft which we can later view in tensor board our output labels which will be the same as our training data folder name and the image directory where we stored our Vader images let's go ahead and run the script right from terminal it'll take about 30 minutes so to train our classifier so do something productive the script should output a training accuracy somewhere between 85 and 99% when it's done and this brings us to our final step we want to write a script that will use our new retrain classifier to detect if a novel image contains Darth Vader we'll write the script ourselves first things first we'll import tensorflow then we'll want to create a variable to store the user input image path we'll create another variable to store the data from that image and one more to load the label of that image from the label file next we'll want to grab our model from the saved retrained graph file store it in the graph def variable and parse it now that we have our image and model ready it's time to make our prediction by feeding the image data into our retrained model to get our prediction output in order to do this we'll create a tensorflow session this will give us an environment to perform operations on our tensor data in the first thing we'll do in a session is get our softmax function tensor from the last layer of our model the softmax function is used in the final layer to map input data into probabilities of an expected output we will execute our softmax tensor function on our input image data via the session run function it will output our predictions as an array we'll next want to sort our prediction labels in order of confidence and lastly for every prediction we have we can get the predicted label and the score and print it out to terminal let's take the script and run it on one of our Vader pictures the result is pretty good tensorflow makes it much easier to classify an image and I've got a challenge for you guys on this episode The Challenge is to create a classifier that you think would be a useful tool for scientists to have it can be any field of science you'd like upload your code to GitHub and in the read me write up a few sentences on how a scientist would use this post a repository in the comment section and I'll judge them based on utility and accuracy the winner gets a shout out from me two videos from now so in 2 weeks and I'll also send you a free signed copy of my book decentralized applications for now I've got to go not buy the iPhone 7 so thanks for watching