Load Data

Load the digit sample data as an image datastore. TheimageDatastorefunction automatically labels the images based on folder names.

digitDatasetPath = fullfile (matlabroot,'toolbox','nnet','nndemos',...'nndatasets','DigitDataset'); imds = imageDatastore(digitDatasetPath,...'IncludeSubfolders',true,...'LabelSource','foldernames');

Divide the data into training and validation data sets, so that each category in the training set contains 750 images, and the validation set contains the remaining images from each label.splitEachLabelsplits the image datastore into two new datastores for training and validation.

numTrainFiles = 750; [imdsTrain,imdsValidation] = splitEachLabel(imds,numTrainFiles,'randomize');

Define Network Architecture

Define the convolutional neural network architecture. Specify the size of the images in the input layer of the network and the number of classes in the fully connected layer before the classification layer. Each image is 28-by-28-by-1 pixels and there are 10 classes.

inputSize = [28 28 1]; numClasses = 10; layers = [ imageInputLayer(inputSize) convolution2dLayer(5,20) batchNormalizationLayer reluLayer fullyConnectedLayer(numClasses) softmaxLayer classificationLayer];

For more information about deep learning layers, seeList of Deep Learning Layers.

Train Network

Specify the training options and train the network.

By default,trainNetworkuses a GPU if one is available, otherwise, it uses a CPU. Training on a GPU requires Parallel Computing Toolbox™ and a supported GPU device. For information on supported devices, seeGPU Support by Release(Parallel Computing Toolbox). You can also specify the execution environment by using the'ExecutionEnvironment'name-value pair argument oftrainingOptions.

options = trainingOptions('sgdm',...“MaxEpochs”,4,...'ValidationData',imdsValidation,...'ValidationFrequency',30,...'Verbose',false,...“阴谋”,'training-progress'); net = trainNetwork(imdsTrain,layers,options);

For more information about training options, seeSet Up Parameters and Train Convolutional Neural Network.

Test Network

Classify the validation data and calculate the classification accuracy.

YPred = classify(net,imdsValidation); YValidation = imdsValidation.Labels; accuracy = mean(YPred == YValidation)

accuracy = 0.9892

For next steps in deep learning, you can try using pretrained network for other tasks. Solve new classification problems on your image data with transfer learning or feature extraction. For examples, seeStart Deep Learning Faster Using Transfer LearningandTrain Classifiers Using Features Extracted from Pretrained Networks. To learn more about pretrained networks, seePretrained Deep Neural Networks.