Resource retrieval
Retrieve the resource object:
Out[1]= | |
Get the pre-trained net:
Out[2]= | |
Basic usage
Classify an image:
Out[3]= | |
The prediction is an Entity object, which can be queried:
Out[4]= | |
Get a list of available properties of the predicted Entity:
Out[5]= | |
Obtain the probabilities of the ten most likely entities predicted by the net:
Out[6]= | |
An object outside the list of the ImageNet classes will be misidentified:
Out[7]= | |
Obtain the list of names of all available classes:
Out[8]= | |
Feature extraction
Remove the last three layers of the trained net so that the net produces a vector representation of an image:
Out[9]= | |
Get a set of images:
Visualize the features of a set of images:
Out[11]= | |
Visualize convolutional weights
Extract the weights of the first convolutional layer in the trained net:
Show the dimensions of the weights:
Out[13]= | |
Visualize the weights as a list of 64 images of size 3x3:
Out[14]= | |
Transfer learning
Use the pre-trained model to build a classifier for telling apart images of dogs and cats. Create a test set and a training set:
Remove the linear layer from the pre-trained net:
Out[17]= | |
Create a new net composed of the pre-trained net followed by a linear layer and a softmax layer:
Out[18]= | |
Train on the dataset, freezing all the weights except for those in the "linearNew" layer (use TargetDevice -> "GPU" for training on a GPU):
Out[19]= | |
Perfect accuracy is obtained on the test set:
Out[20]= | |
Net information
Inspect the number of parameters of all arrays in the net:
Out[21]= | |
Obtain the total number of parameters:
Out[22]= | |
Obtain the layer type counts:
Out[23]= | |
Display the summary graphic:
Out[24]= | |
Export to MXNet
Export the net into a format that can be opened in MXNet:
Out[25]= | |
Export also creates a net.params file containing parameters:
Out[26]= | |
Get the size of the parameter file:
Out[27]= | |
The size is similar to the byte count of the resource object:
Out[28]= | |
Represent the MXNet net as a graph:
Out[29]= | |