-
大小: 370KB文件類型: .zip金幣: 2下載: 0 次發(fā)布日期: 2021-06-02
- 語(yǔ)言: Matlab
- 標(biāo)簽: matlab??遷移學(xué)習(xí)??深度學(xué)習(xí)??神經(jīng)網(wǎng)絡(luò)??圖像識(shí)別??
資源簡(jiǎn)介
基于深度遷移學(xué)習(xí)的小樣本圖像分類matlab程序,網(wǎng)絡(luò)模型基于AlexNet,文件包含了圖像數(shù)據(jù)集,輸出結(jié)果可靠。

代碼片段和文件信息
%?加載數(shù)據(jù)
unzip(‘MerchData.zip‘);
imds?=?imageDatastore(‘MerchData‘...
????‘IncludeSubfolders‘true...
????‘LabelSource‘‘foldernames‘);
%?劃分?jǐn)?shù)據(jù)集
[imdsTrainimdsValidation]?=?splitEachLabel(imds0.7‘randomized‘);
%?顯示數(shù)據(jù)集的一些圖像
numTrainImages?=?numel(imdsTrain.Labels);
idx?=?randperm(numTrainImages16);
figure
for?i?=?1:16
????subplot(44i)
????I?=?readimage(imdsTrainidx(i));
????imshow(I)
end
%?加載預(yù)訓(xùn)練網(wǎng)絡(luò)
net?=?alexnet;
analyzeNetwork(net)
inputSize?=?net.layers(1).InputSize
%?替換最終層
layersTransfer?=?net.layers(1:end-3);
numClasses?=?numel(categories(imdsTrain.Labels))
layers?=?[
????layersTransfer
????fullyConnectedlayer(numClasses‘WeightLearnRateFactor‘20‘BiasLearnRateFactor‘20)
????softmaxlayer
????classificationlayer];
%?開始訓(xùn)練網(wǎng)絡(luò)
pixelRange?=?[-30?30];
imageAugmenter?=?imageDataAugmenter(?...
????‘RandXReflection‘true?...
????‘RandXTranslation‘pixelRange?...
????‘RandYTranslation‘pixelRange);
augimdsTrain?=?augmentedImageDatastore(inputSize(1:2)imdsTrain?...
????‘DataAugmentation‘imageAugmenter);
augimdsValidation?=?augmentedImageDatastore(inputSize(1:2)imdsValidation);
options?=?trainingOptions(‘sgdm‘?...
????‘MiniBatchSize‘10?...
????‘MaxEpochs‘6?...
????‘InitialLearnRate‘1e-4?...
????‘Shuffle‘‘every-epoch‘?...
????‘ValidationData‘a(chǎn)ugimdsValidation?...
????‘ValidationFrequency‘3?...
????‘Verbose‘false?...
????‘Plots‘‘training-progress‘);
netTransfer?=?trainNetwork(augimdsTrainlayersoptions);
%?對(duì)圖像進(jìn)行分類
idx?=?randperm(numel(imdsValidation.Files)4);
figure
for?i?=?1:4
????subplot(22i)
????I?=?readimage(imdsValidationidx(i));
????imshow(I)
????label?=?YPred(idx(i));
????title(string(label));
end
%?分類準(zhǔn)確度
YValidation?=?imdsValidation.Labels;
accuracy?=?mean(YPred?==?YValidation)
?屬性????????????大小?????日期????時(shí)間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2019-08-25?10:03??code\
?????文件????????1844??2019-07-25?19:47??code\AlexNet.m
?????目錄???????????0??2019-08-25?10:03??code\MerchData\
?????目錄???????????0??2019-08-25?10:03??code\MerchData\MathWorks?Cap\
?????文件????????6070??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_0.jpg
?????文件????????4472??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_123.jpg
?????文件????????3997??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_148.jpg
?????文件????????5023??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_173.jpg
?????文件????????5427??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_198.jpg
?????文件????????5410??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_222.jpg
?????文件????????5003??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_247.jpg
?????文件????????4696??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_272.jpg
?????文件????????4220??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_297.jpg
?????文件????????5266??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_321.jpg
?????文件????????5436??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_346.jpg
?????文件????????5160??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_371.jpg
?????文件????????5801??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_49.jpg
?????文件????????5136??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_74.jpg
?????文件????????4819??2016-12-07?13:41??code\MerchData\MathWorks?Cap\Hat_99.jpg
?????目錄???????????0??2019-08-25?10:03??code\MerchData\MathWorks?Cube\
?????文件????????6609??2016-12-07?13:41??code\MerchData\MathWorks?Cube\MathWorks?cube_0.jpg
?????文件????????6539??2016-12-07?13:41??code\MerchData\MathWorks?Cube\MathWorks?cube_1099.jpg
?????文件????????4746??2016-12-07?13:41??code\MerchData\MathWorks?Cube\MathWorks?cube_120.jpg
?????文件????????7362??2016-12-07?13:41??code\MerchData\MathWorks?Cube\MathWorks?cube_1300.jpg
?????文件????????5050??2016-12-07?13:41??code\MerchData\MathWorks?Cube\MathWorks?cube_1402.jpg
?????文件????????6531??2016-12-07?13:41??code\MerchData\MathWorks?Cube\MathWorks?cube_1503.jpg
?????文件????????5911??2016-12-07?13:41??code\MerchData\MathWorks?Cube\MathWorks?cube_189.jpg
?????文件????????5304??2016-12-07?13:41??code\MerchData\MathWorks?Cube\MathWorks?cube_29.jpg
?????文件????????6614??2016-12-07?13:41??code\MerchData\MathWorks?Cube\MathWorks?cube_390.jpg
?????文件????????7670??2016-12-07?13:41??code\MerchData\MathWorks?Cube\MathWorks?cube_492.jpg
?????文件????????5494??2016-12-07?13:41??code\MerchData\MathWorks?Cube\MathWorks?cube_593.jpg
............此處省略52個(gè)文件信息
評(píng)論
共有 條評(píng)論