svm hog图像分类

%读取训练集,测试集
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参考

猜你喜欢

转载自blog.csdn.net/qq_41244435/article/details/86560667