京东金融大数据竞赛猪脸识别（3）- 图像特征提取之二

clear;
trainPath = fullfile(pwd,'image');
trainData = imageDatastore(trainPath,...
'IncludeSubfolders',true,'LabelSource','foldernames');

[trainingImages,testImages] = splitEachLabel(trainData,0.7,'randomized');
numTrainImages = numel(trainingImages.Labels);
%加载预训练模型
net = alexnet;
%指定用来提取特征的层
layer = 'fc7';
%提取指定层训练数据特征
trainingFeatures = activations(net,trainingImages,layer);
%提取指定层测试数据特征
testFeatures = activations(net,testImages,layer);
%获取训练数据标签
trainingLabels = trainingImages.Labels;
%获取测试数据标签
testLabels = testImages.Labels;
save('alexnetFeature.mat','trainingFeatures','trainingLabels','trainingFeatures','testLabels');