www.gusucode.com > IPCV_Eval_Kit_R2019a_0ce6858工具箱matlab程序源码 > IPCV_Eval_Kit_R2019a_0ce6858/code/demo_files/I5_04_2_myHOGDigitClassification_KNN_R2015a.m

    %% HOG (Histogram of Oriented Gradient) 摿挜検 偲
%  k嵟嬤朤暘椶婍丗KNN (k-nearest neighbor) classifier 傪巊偭偨丄庤彂偒悢帤偺幆暿
clear;clc;close all;imtool close all

% 僩儗乕僯儞僌夋憸乮101枃x10暥帤庬乯偲僥僗僩夋憸乮12枃x10暥帤庬乯傊偺愨懳僷僗傪愝掕
pathData = [toolboxdir('vision'), '\visiondata\digits']
trainSet = imageSet([pathData,'\synthetic'  ], 'recursive');
testSet  = imageSet([pathData,'\handwritten'], 'recursive');

%% 慡僩儗乕僯儞僌梡夋憸椺偺昞帵
figure;montage([trainSet.ImageLocation], 'Size', [26 40]);

%% 慡僥僗僩夋憸傪儌儞僞乕僕儏昞帵 (12枃 x 10暥帤庬丗奺庤彂偒悢帤傪擣幆)
figure;montage([testSet(:).ImageLocation], 'Size', [10,12]);

%% 4x4偺僙儖僒僀僘傪巊梡 (324師尦儀僋僩儖)
cellSize = [4 4];
hogFeatureSize = 324;                   % length(hog_4x4)

%% [KNN暘椶婍偺峔抸]丗fitcknn傪巊梡
% 10暥帤暘偺trainingFeatures 傪奿擺偡傞攝楍傪偁傜偐偠傔嶌惢
trainingFeatures  = zeros(10*101,hogFeatureSize, 'single');
trainingLabels    = zeros(10*101,1);

% HOG摿挜検傪拪弌
for digit = 0:9   % 暥帤'0'乣'9'
  for i = 1:101         % 奺悢帤偛偲偵101枃偺僩儗乕僯儞僌梡夋憸
    img = read(trainSet(digit+1), i);  %僩儗乕僯儞僌夋憸偺撉崬傒       trainSet()偼丄1偐傜巒傑傞偺偱丄+1
    img = im2bw(img,graythresh(img));   % 擇抣壔
             
    trainingFeatures((digit)*101+i,:) = extractHOGFeatures(img,'CellSize',cellSize);
    trainingLabels((digit)*101+i)     = digit;
  end
end
% KNN暘椶婍偺妛廗 (k=5)
knnModel5 = fitcknn(trainingFeatures, trainingLabels, 'NumNeighbors', 5)    %嬤偄傕偺5偮傪庢傝懡悢寛

%% [幆暿] 嶌惉偟偨暘椶婍偱庤彂偒悢帤(120枃)傪幆暿\帵丗predict()
Ir = zeros([16,16,3,120], 'uint8');      % 寢壥傪奿擺偡傞攝楍
cntTrue = 0;
for digit = 0:9   % 
  for i = 1:12         % 奺悢帤偛偲偵12枃偺庤彂偒暥帤
    img = read(testSet(digit+1), i);    % testSet()偼丄1偐傜巒傑傞偺偱丄+1
    BW = im2bw(img,graythresh(img));    % 2抣壔

    testFeatures = extractHOGFeatures(BW,'CellSize',cellSize);
    predictedNum = predict(knnModel5, testFeatures);           % testFeature 傪攝楍偵偟偰丄偁偲偱傑偲傔偰敾掕傕壜
    
    if predictedNum == digit    %惓偟偄幆暿偼惵怓丄岆擣幆偼愒怓
      Ir(:,:,:,digit*12+i) = insertText(img,[6 4],num2str(predictedNum),'FontSize',9,'TextColor','blue','BoxOpacity',0.4);
      cntTrue = cntTrue+1;
    else
      Ir(:,:,:,digit*12+i) = insertText(img,[6 4],num2str(predictedNum),'FontSize',9,'TextColor','red','BoxOpacity',0.4); 
    end 

  end
end
% 寢壥偺昞帵
figure;montage(Ir, 'Size', [10,12]); title(['Correct Prediction: ' num2str(cntTrue)]);

%%








%% Copyright 2013-2014 The MathWorks, Inc.
% 夋憸僨乕僞僙僢僩
% 僩儗乕僯儞僌夋憸丗insertText娭悢偱帺摦嶌惉 (廃埻偵暿偺悢帤桳傝)
% 僥僗僩夋憸丗庤彂偒偺夋憸傪巊梡