Eye Blink Detection - Training

clear;clc;close all;

fea=[];
group=[];


dos('attrib -h -r -s /S .\trainset\Thumbs.db');
dos('del /S .\trainset\Thumbs.db');

folder=dir('.\trainset');
count=0;

LeftEyeDetector = vision.CascadeObjectDetector('LeftEyeCART');


%%

for mn=3:length(folder)
    count=count+1;
    address=strcat('.\trainset\',folder(mn).name);
    
    files=dir(address);
    num=numel(files);
    
    for i=3:num
        str=strcat('.\trainset\',folder(mn).name,'\',files(i).name);
        
        msg_str = sprintf('Processing Folder %d, File %d\n',count,i-2);
        fprintf(msg_str);
        
        im=imread(str);
        im = imresize(im,[500,NaN]);
        
        [r,c,p] = size(im);
        
        if p == 3
            gr = rgb2gray(im);
        else
            gr = im;
        end
        
        gr = imadjust(gr);
        gr = histeq(gr);
        
        bboxleye = step(LeftEyeDetector, gr);
        
        if numel(bboxleye(1,:))== 4
            
            LeftEye=imcrop(gr,bboxleye(1,:));
            LeftEye=imresize(LeftEye,[100 100]);
            pause(0.01)
            
            [feature] = extractHOGFeatures(LeftEye);
            fea=[fea;feature];
            group=[group;count];
            
            imshow(LeftEye);
            
        end
    
    end
end


warndlg('Training Complete !');


save EyeDatabase fea group