Main Content

mobilenetv2

MobileNet-v2 convolutional neural network

Description

MobileNet-v2 is a convolutional neural network that is 53 layers deep. You can load a pretrained version of the network trained on more than a million images from the ImageNet database [1]. The pretrained network can classify images into 1000 object categories, such as keyboard, mouse, pencil, and many animals. As a result, the network has learned rich feature representations for a wide range of images. The network has an image input size of 224-by-224. For more pretrained networks in MATLAB®, see Pretrained Deep Neural Networks.

You can use classify to classify new images using the MobileNet-v2 model. Follow the steps of Classify Image Using GoogLeNet and replace GoogLeNet with MobileNet-v2.

To retrain the network on a new classification task, follow the steps of Train Deep Learning Network to Classify New Images and load MobileNet-v2 instead of GoogLeNet.

example

net = mobilenetv2 returns a MobileNet-v2 network trained on the ImageNet data set.

This function requires the Deep Learning Toolbox™ Model for MobileNet-v2 Network support package. If this support package is not installed, then the function provides a download link.

net = mobilenetv2('Weights','imagenet') returns a MobileNet-v2 network trained on the ImageNet data set. This syntax is equivalent to net = mobilenetv2.

lgraph = mobilenetv2('Weights','none') returns the untrained MobileNet-v2 network architecture. The untrained model does not require the support package.

Examples

collapse all

Download and install the Deep Learning Toolbox Model for MobileNet-v2 Network support package.

Type mobilenetv2 at the command line.

mobilenetv2

If the Deep Learning Toolbox Model for MobileNet-v2 Network support package is not installed, then the function provides a link to the required support package in the Add-On Explorer. To install the support package, click the link, and then click Install. Check that the installation is successful by typing mobilenetv2 at the command line. If the required support package is installed, then the function returns a DAGNetwork object.

mobilenetv2
ans = 

  DAGNetwork with properties:

         Layers: [155×1 nnet.cnn.layer.Layer]
    Connections: [164×2 table]

Output Arguments

collapse all

Pretrained MobileNet-v2 convolutional neural network, returned as a DAGNetwork object.

Untrained MobileNet-v2 convolutional neural network architecture, returned as a LayerGraph object.

References

[1] ImageNet. http://www.image-net.org

[2] Sandler, M., Howard, A., Zhu, M., Zhmoginov, A. and Chen, L.C. "MobileNetV2: Inverted Residuals and Linear Bottlenecks." In 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition (pp. 4510-4520). IEEE.

Extended Capabilities

Introduced in R2019a