|
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); |