Name: powerai-transfer-learning
Owner: International Business Machines
Description: Image recognition training with TensorFlow Inception and transfer learning
Created: 2017-07-06 20:27:20.0
Updated: 2018-05-23 10:59:16.0
Pushed: 2017-12-01 17:28:09.0
Homepage: https://developer.ibm.com/code/patterns/image-recognition-training-powerai-notebooks/
Size: 24130
Language: Python
GitHub Committers
User | Most Recent Commit | # Commits |
---|
Other Committers
User | Most Recent Commit | # Commits |
---|
Read this in other languages: ???.
Transfer learning is the process of taking a pre-trained model (the weights and parameters of a network that has been trained on a large dataset by somebody else) and ?fine-tuning? the model with your own dataset. The idea is that this pre-trained model will act as a feature extractor. You will remove the last layer of the network and replace it with your own classifier (depending on what your problem space is). You then freeze the weights of all the other layers and train the network normally (Freezing the layers means not changing the weights during gradient descent/optimization). For this experiment we used Google's Inception-V3 pretrained model for Image Classification. This model consists of two parts:
The pre-trained Inception-v3 model achieves state-of-the-art accuracy for recognizing general objects with 1000 classes. The model extracts general features from input images in the first part and classifies them based on those features in the second part. We will use this pre-trained model and re-train it it to classify houses with or without swimming pools.
Follow these steps to setup and run this Code Pattern. The steps are described in detail below.
IBM has partnered with Nimbix to provide cognitive developers a trial account that provides 24-hours of free processing time on the PowerAI platform. Follow these steps to register for access to Nimbix to try the PowerAI Cognitive Code Patterns and explore the platform.
Go to the IBM Marketplace PowerAI Portal, and click Request Trial
.
On the IBM PowerAI Trial page, shown below, enter the required information to sign up for an IBM account and click Continue
. If you already have an IBM ID, click Already have an account? Log in
, enter your credentials and click Continue
.
On the Almost there? page, shown below, enter the required information and click Continue
to complete the registration and launch the IBM Marketplace Products and Services page.
Your IBM Marketplace Products and Services page displays all offerings that are available to you; the PowerAI Trial should now be one of them. From the PowerAI Trial section, click Launch
, as shown below, to launch the IBM PowerAI trial page.
The Welcome to IBM PowerAI Trial page provides instructions for accessing the trial, as shown below. Alternatively, you will receive an email confirming your registration with similar instructions that you can follow to start the trial.
Summary of steps for starting the trial:
Start a terminal session from your local machine and issue the following command where {IP Address}
is the IP Address (or host name) shown on the welcome page (or in the confirmation email).
-L 8888:localhost:8888 nimbix@{IP Address}
Enter the password shown on the welcome page (or in the confirmation email) when prompted.
From your local browser, go to the following URL to get started: http://localhost:8888/tree/.
Use git clone to download the example notebook, dataset, and retraining library with a single command.
`New
pull-down and selecting `
Terminal``.clone https://github.com/IBM/powerai-transfer-learning
Files
tab and click on powerai-transfer-learning
then notebooks
and then Classifying-House-And-Pool-Images.ipynb
to open the notebook.When a notebook is executed, what is actually happening is that each code cell in the notebook is executed, in order, from top to bottom.
Each code cell is selectable and is preceded by a tag in the left margin. The tag
format is In [x]:
. Depending on the state of the notebook, the x
can be:
*
, this indicates that the cell is currently executing.There are several ways to execute the code cells in your notebook:
Play
button in the toolbar.Cell
menu bar, there are several options available. For example, you
can Run All
cells in your notebook, or you can Run All Below
, that will
start executing from the first cell under the currently selected cell, and then
continue executing all cells that follow.When you run the “Main” code cell you can watch the training as the accuracy quickly improves. At the end, the final test accuracy is shown. We usually see somewhere around 85% accuracy with these images.
We captured the model before and after the training. Look at our example images at the bottom of the notebook and see our before and after results.
The results should go from no recognition ability at all to a pretty good success rate. You might find it interesting to try different images and see if you can identify why it has more difficulty classifying some images.
Because this notebook is running temporarily on a Nimbix Cloud server, use the following options to save your work:
Under the File
menu, there are options to:
Download as...
will download the notebook to your local system.Print Preview
will allow you to print the current state of the
notebook.When you are done with your work, please cancel your subscription by issuing the following command in your ssh session or by visiting the Manage
link on the My Products and Services page.
poweroff --force