mobilenetv2

R2026b

(Not recommended) MobileNet-v2 convolutional neural network

• 

mobilenetv2 is not recommended. Use the imagePretrainedNetwork function instead and specify the "mobilenetv2" model. For more information, see Version History.

To learn more about how to transition trainNetwork, SeriesNetwork, and DAGNetwork code to dlnetwork workflows, see Transition trainNetwork, SeriesNetwork, and DAGNetwork Code to dlnetwork Workflows.

Syntax

net = mobilenetv2

net = mobilenetv2('Weights','imagenet')

lgraph = mobilenetv2('Weights','none')

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.

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

Download MobileNet-v2 Support Package

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]

Visualize the network using Deep Network Designer.

deepNetworkDesigner(mobilenetv2)

Explore other pretrained neural networks in Deep Network Designer by clicking New.

If you need to download a neural network, pause on the desired neural network and click Install to open the Add-On Explorer.

Output Arguments

net — Pretrained MobileNet-v2 convolutional neural network

DAGNetwork object

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

lgraph — Untrained MobileNet-v2 convolutional neural network architecture

LayerGraph object

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

References

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

[2] Sandler, Mark, Andrew Howard, Menglong Zhu, Andrey Zhmoginov, and Liang-Chieh Chen. “MobileNetV2: Inverted Residuals and Linear Bottlenecks.” In 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, 4510–20. Salt Lake City, UT: IEEE, 2018. https://doi.org/10.1109/CVPR.2018.00474.

Extended Capabilities

C/C++ Code Generation

Generate C and C++ code using MATLAB® Coder™.

For code generation, you can load the network by using the syntax net = mobilenetv2 or by passing the mobilenetv2 function to coder.loadDeepLearningNetwork (MATLAB Coder). For example: net = coder.loadDeepLearningNetwork('mobilenetv2')

For more information, see Load Pretrained Networks for Code Generation (MATLAB Coder).

The syntax mobilenetv2('Weights','none') is not supported for code generation.

GPU Code Generation

Generate CUDA® code for NVIDIA® GPUs using GPU Coder™.

Usage notes and limitations:

• For code generation, you can load the network by using the syntax net = mobilenetv2 or by passing the mobilenetv2 function to coder.loadDeepLearningNetwork (GPU Coder). For example: net = coder.loadDeepLearningNetwork('mobilenetv2')

  For more information, see Load Pretrained Networks for Code Generation (GPU Coder).

• The syntax mobilenetv2('Weights','none') is not supported for GPU code generation.

Version History

Introduced in R2019a

R2024a: Not Recommended

mobilenetv2 is not recommended. Use the imagePretrainedNetwork function instead and specify "mobilenetv2" as the model.

There are no plans to remove support for the mobilenetv2 function. However, the imagePretrainedNetwork function has additional functionality that helps with transfer learning workflows. For example, you can specify the number of classes in your data using the numClasses option, and the function returns a network that is ready for retraining without the need for modification.

The imagePretrainedNetwork function returns the network as a dlnetwork object, which does not store the class names, To get the class names of the pretrained network, use the second output argument of the imagePretrainedNetwork function.

This table shows some typical usages of the mobilenetv2 function and how to update your code to use the imagePretrainedNetwork function instead.

Not Recommended	Recommended
net = mobilenetv2;	[net,classNames] = imagePretrainedNetwork("mobilenetv2");
net = mobilenetv2(Weights="none");	net = imagePretrainedNetwork("mobilenetv2",Weights="none");

The imagePretrainedNetwork returns a dlnetwork object, which also has these advantages:

• dlnetwork objects are a unified data type that supports network building, prediction, built-in training, visualization, compression, verification, and custom training loops.

• dlnetwork objects support a wider range of network architectures that you can create or import from external platforms.

• The trainnet function supports dlnetwork objects, which enables you to easily specify loss functions. You can select from built-in loss functions or specify a custom loss function.

• Training and prediction with dlnetwork objects is typically faster than LayerGraph and trainNetwork workflows.

To train a neural network specified as a dlnetwork object, use the trainnet function.

To learn more about how to transition trainNetwork, SeriesNetwork, and DAGNetwork code to dlnetwork workflows, see Transition trainNetwork, SeriesNetwork, and DAGNetwork Code to dlnetwork Workflows.