Skip to content
Snippets Groups Projects
Select Git revision
  • master
1 result

goNCD.m

Blame
  • goNCD.m 1.33 KiB
    
    if isempty(gcp('nocreate'))
        parpool(36)
    end
    ROOT = 'F:\Raw\Images\Temple\RPE\2014-01-24\'
    tic
    Trellis=[];
    for category=1:5
            droot = [ROOT 'Category ' num2str(category) '\'];
            flist = dir([droot '*.tif']);
            for ff=1:length(flist)
                
                im = imread([droot flist(ff).name]);            
                im = rgb2gray(im);
    %             im=im2double(im);
                
    %             im = CudaMex('ContrastEnhancement', im, [5 5 1],[3 3 1]);
    %             im = mat2gray(stdfilt(im, getnhood(strel('disk',5))));
    %             
                
                im=gpuArray(im);           
                im=PreProcess(im);
                im=gather(im);
                [m n] = size(im);
                im = im(1:m/2, 1:n/2);
                nt=[];
                nt.im=im;
                nt.category = category;
                nt.bCount=-1;
                Trellis=[Trellis nt];
            end
    end
    
    parfor i=1:length(Trellis)
        Trellis(i).bCount = getCount(Trellis(i).im,i);
    end
    
    
    d=[];
    parfor i=1:length(Trellis)
        tTrain = Trellis;
        tTrain(i)=[];
        tTest = Trellis(i);    
        for nClass=1:5
            idx = find([tTrain.category]==nClass);
            d(i,nClass)=NCD([tTrain(idx) tTest],i)-NCD([tTrain(idx)],i);
        end
    end
                
    [mm idx]=min(d,[],2);            
    length(find(idx==[Trellis.category]'))
    
    toc