Create new deep networks for image classification and regression tasks by defining the network architecture and training the network from scratch. You can also use transfer learning to take advantage of the knowledge provided by a pretrained network to learn new patterns in new data. Fine-tuning a pretrained image classification network with transfer learning is typically much faster and easier than training from scratch. Using pretrained deep networks enables you to quickly learn new tasks without defining and training a new network, having millions of images, or having a powerful GPU.
After defining the network architecture, you must define training
parameters using the
trainingOptions function. You
can then train the network using
trainNetwork. Use the trained
network to predict class labels or numeric responses.
You can train a convolutional neural network on a CPU, a GPU, multiple
CPUs or GPUs, or in parallel on a cluster or in the cloud. Training on a GPU
or in parallel requires Parallel Computing Toolbox™. Using a GPU requires a CUDA® enabled NVIDIA® GPU with compute capability 3.0 or higher. Specify the
execution environment using the
|Deep Network Designer||Design, visualize, and train deep learning networks|
|SqueezeNet convolutional neural network|
|GoogLeNet convolutional neural network|
|Inception-v3 convolutional neural network|
|DenseNet-201 convolutional neural network|
|MobileNet-v2 convolutional neural network|
|ResNet-18 convolutional neural network|
|ResNet-50 convolutional neural network|
|ResNet-101 convolutional neural network|
|Xception convolutional neural network|
|Pretrained Inception-ResNet-v2 convolutional neural network|
|Pretrained NASNet-Large convolutional neural network|
|Pretrained NASNet-Mobile convolutional neural network|
|Pretrained ShuffleNet convolutional neural network|
|DarkNet-19 convolutional neural network|
|DarkNet-53 convolutional neural network|
|EfficientNet-b0 convolutional neural network|
|AlexNet convolutional neural network|
|VGG-16 convolutional neural network|
|VGG-19 convolutional neural network|
|Image input layer|
|3-D image input layer|
|Feature input layer|
|2-D convolutional layer|
|3-D convolutional layer|
|2-D grouped convolutional layer|
|Transposed 2-D convolution layer|
|Transposed 3-D convolution layer|
|Fully connected layer|
|Rectified Linear Unit (ReLU) layer|
|Leaky Rectified Linear Unit (ReLU) layer|
|Clipped Rectified Linear Unit (ReLU) layer|
|Exponential linear unit (ELU) layer|
|Hyperbolic tangent (tanh) layer|
|Batch normalization layer|
|Group normalization layer|
|Channel-wise local response normalization layer|
|2-D crop layer|
|3-D crop layer|
|Average pooling layer|
|3-D average pooling layer|
|Global average pooling layer|
|3-D global average pooling layer|
|Global max pooling layer|
|3-D global max pooling layer|
|Max pooling layer|
|3-D max pooling layer|
|Max unpooling layer|
|Depth concatenation layer|
|Graph of network layers for deep learning|
|Plot neural network layer graph|
|Add layers to layer graph|
|Remove layers from layer graph|
|Replace layer in layer graph|
|Connect layers in layer graph|
|Disconnect layers in layer graph|
|Directed acyclic graph (DAG) network for deep learning|
|Classify data using a trained deep learning neural network|
|Predict responses using a trained deep learning neural network|
|Compute deep learning network layer activations|
|Create confusion matrix chart for classification problem|
|Sort classes of confusion matrix chart|
|ConfusionMatrixChart Properties||Confusion matrix chart appearance and behavior|
This example shows how to classify an image using the pretrained deep convolutional neural network GoogLeNet.
This example shows how to classify images from a webcam in real time using the pretrained deep convolutional neural network GoogLeNet.
Interactively fine-tune a pretrained deep learning network to learn a new image classification task.
This example shows how to use transfer learning to retrain a convolutional neural network to classify a new set of images.
This example shows how to extract learned image features from a pretrained convolutional neural network and use those features to train an image classifier.
This example shows how to fine-tune a pretrained GoogLeNet convolutional neural network to perform classification on a new collection of images.
Learn how to download and use pretrained convolutional neural networks for classification, transfer learning and feature extraction.
This example shows how to create and train a simple convolutional neural network for deep learning classification.
Interactively build and edit deep learning networks.
This example shows how to fit a regression model using convolutional neural networks to predict the angles of rotation of handwritten digits.
Discover all the deep learning layers in MATLAB®.
Learn about the layers of a convolutional neural network (ConvNet), and the order they appear in a ConvNet.
Generate MATLAB code to recreate designing and training a network in Deep Network Designer.
This example shows how to create a deep learning neural network with residual connections and train it on CIFAR-10 data.
This example shows how to create and train a simple neural network for deep learning feature data classification.
Learn how to define and train deep learning networks with multiple inputs or multiple outputs.
This example shows how to train a generative adversarial network (GAN) to generate images.
This example shows how to train a conditional generative adversarial network (CGAN) to generate images.
This example shows how to train a network to transfer the style of an image to a second image.
This example shows how to train a deep learning model for image captioning using attention.
This example shows how to train a network that classifies handwritten digits with a custom learning rate schedule.
This example shows how to train a deep learning network with multiple outputs that predict both labels and angles of rotations of handwritten digits.
This example shows how to train a Siamese network to identify similar images of handwritten characters.
This example shows how to import a custom weighted classification layer and add it to a pretrained network in Deep Network Designer.
This example shows how to use Deep Network Designer to construct and train an image-to-image regression network for super resolution.
Discover deep learning capabilities in MATLAB using convolutional neural networks for classification and regression, including pretrained networks and transfer learning, and training on GPUs, CPUs, clusters, and clouds.
Learn how to set up training parameters for a convolutional neural network.
Learn how to resize images for training, prediction, and classification, and how to preprocess images using data augmentation, transformations, and specialized datastores.
Read and preprocess volumetric image and label data for 3-D deep learning.
Learn how to use datastores in deep learning applications.
This example shows how to convert a trained classification network into a regression network.
Learn how to improve the accuracy of deep learning networks.
Discover data sets for various deep learning tasks.
Import and visualize data in Deep Network Designer.