Main Content

Image Classifier

Classify data using a trained deep learning neural network

  • Library:
  • Deep Learning Toolbox / Deep Neural Networks

  • Image classifier block

Description

The Image Classifier block predicts class labels for the data at the input by using the trained network specified through the block parameter. This block allows loading of a pretrained network into the Simulink® model from a MAT-file or from a MATLAB® function.

Note

The Image Classifier block does not support sequence networks and multiple input and multiple output networks (MIMO).

Ports

Input

expand all

A h-by-w-by-c-by-N numeric array, where h, w, and c are the height, width, and number of channels of the images, respectively, and N is the number of images. If the array contains NaNs, then they are propagated through the network.

Output

expand all

Predicted class labels with the highest score, returned as a N-by-1 enumerated vector of labels, where N is the number of observations.

Predicted scores, returned as a N-by-K matrix, where N is the number of observations, and K is the number of classes.

Labels associated with the predicted scores, returned as a N-by-K matrix, where N is the number of observations, and K is the number of classes.

Parameters

expand all

Specify the source for the trained network. Select one of the following:

  • Network from MAT-file— Import a trained network from a MAT-file containing a SeriesNetwork or a DAGNetwork object.

  • Network from MATLAB function— Import a pretrained network from a MATLAB function. For example, by using the googlenet function.

This parameter specifies the name of the MAT-file that contains the trained deep learning network to load. If the file is not on the MATLAB path, use the Browse button to locate the file.

Dependencies

To enable this parameter, set the Network parameter to Network from MAT-file.

This parameter specifies the name of the MATLAB function for the pretrained deep learning network. For example, use googlenet function to import the pretrained GoogLeNet model.

Dependencies

To enable this parameter, set the Network parameter to Network from MATLAB function.

Size of mini-batches to use for prediction, specified as a positive integer. Larger mini-batch sizes require more memory, but can lead to faster predictions.

Resize the data at the input port to the input size of the network.

Enable output port ypred that outputs the label with the highest score.

Enable output ports scores and labels that output all predicted scores and associated class labels.

Extended Capabilities

See Also

Introduced in R2020b