%读取训练集,测试集
imdstrain=imageDatastore('E:\svm\train_images','IncludeSubfolders',true,'LabelSource','foldernames');
imdstest=imageDatastore('E:\svm\test_image');
%计算训练集数目
train_disp=countEachLabel(imdstrain);
disp(train_disp);
%获取训练集hog特征,single
imagesize=[256, 256];
image1=readimage(imdstrain,1);
scalimage=imresize(image1,imagesize);
%scalimage=rgb2gray(scalimage);
[features,visualization]=extractHOGFeatures(scalimage);
imshow(scalimage);hold on;plot(visualization);
numImages=length(imdstrain.Files);
featurestrain=zeros(numImages,size(features,2),'single');
for i=1:numImages
imagetrain=readimage(imdstrain,i);
imagetrain=imresize(imagetrain,imagesize);
%imagetrain=rgb2gray(imagetrain);
featurestrain(i,:)=extractHOGFeatures(imagetrain);
end
%对训练集添标签
trainlabels=imdstrain.Labels;
%分类,多分类用fitcecoc,fitcsvm用于二分类
classifer=fitcecoc(featurestrain,trainlabels);
%测试
numtest=length(imdstest.Files);
for i=1:numtest
testimage=readimage(imdstest,i);
scaltestimage=imresize(testimage,imagesize);
%scaltestimage=rgb2gray(scalimage);
feturetest=extractHOGFeatures(scaltestimage);
[predictIndex,score]=predict(classifer,feturetest);
figure;imshow(scaltestimage);
title(['predictimage ',char(predictIndex)]);
end
https://pan.baidu.com/s/1i5OhC7z密码: utn7
https://blog.csdn.net/zouxy09/article/details/7929348 hog参考