a b/functions/fineTuneCNN.m 

  1
  
function [netTransfer] = fineTuneCNN(imagesCell, Labels, folder, inputSize, imageAugmenter, layers, options)





  
  

  2
  





  
  

  3
  
% imdsTrain = imageDatastore(folder, 'IncludeSubfolders', true, 'LabelSource','foldernames');





  
  

  4
  





  
  

  5
  
im_temp = imagesCell{1};





  
  

  6
  
imsizeOrig = size(im_temp);





  
  

  7
  
imArray = zeros(imsizeOrig(1), imsizeOrig(2), imsizeOrig(3), numel(imagesCell));





  
  

  8
  
for ind_im = 1 : numel(imagesCell)





  
  

  9
  
    imArray(:,:,:,ind_im) = imagesCell{ind_im};





  
  

  10
  
end %for ind_im





  
  

  11
  





  
  

  12
  
% whos imArray Labels





  
  

  13
  





  
  

  14
  
% augimdsTrainl = augmentedImageDatastore(inputSize(1:2), imdsTrain, 'DataAugmentation', imageAugmenter);





  
  

  15
  
% augimdsTrainl = augmentedImageDatastore(inputSize(1:2), imdsTrain);





  
  

  16
  
augimdsTrainl = augmentedImageDatastore(inputSize(1:2), imArray, categorical(Labels)', 'DataAugmentation', imageAugmenter);





  
  

  17
  





  
  

  18
  
netTransfer = trainNetwork(augimdsTrainl, layers, options);