[1422d3]: / functions / fineTuneCNN.m

Download this file

18 lines (13 with data), 822 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
function [netTransfer] = fineTuneCNN(imagesCell, Labels, folder, inputSize, imageAugmenter, layers, options)
% imdsTrain = imageDatastore(folder, 'IncludeSubfolders', true, 'LabelSource','foldernames');
im_temp = imagesCell{1};
imsizeOrig = size(im_temp);
imArray = zeros(imsizeOrig(1), imsizeOrig(2), imsizeOrig(3), numel(imagesCell));
for ind_im = 1 : numel(imagesCell)
imArray(:,:,:,ind_im) = imagesCell{ind_im};
end %for ind_im
% whos imArray Labels
% augimdsTrainl = augmentedImageDatastore(inputSize(1:2), imdsTrain, 'DataAugmentation', imageAugmenter);
% augimdsTrainl = augmentedImageDatastore(inputSize(1:2), imdsTrain);
augimdsTrainl = augmentedImageDatastore(inputSize(1:2), imArray, categorical(Labels)', 'DataAugmentation', imageAugmenter);
netTransfer = trainNetwork(augimdsTrainl, layers, options);