0001 function WBCdetect_ADA(iFolder)
0002 iPath = 'C:\Research\DataSets\';
0003 cd(strcat(iPath,iFolder));
0004 iFiles = dir('MN/*.png');
0005
0006 nFiles = 1;
0007 mkdir('DT');
0008 frames = [];
0009
0010
0011 conf_map = [];
0012
0013 k = 45;
0014 kh = floor(k/2);
0015 k2 = 2*k-1;
0016 kh2= floor(k2/2);
0017 rm_layer = 2;
0018
0019 [rm_k] = create_radial_mean_kernel(k,rm_layer);
0020
0021
0022
0023
0024
0025
0026
0027
0028
0029
0030
0031
0032
0033
0034
0035
0036
0037
0038
0039
0040
0041
0042 disp('Training...');
0043 file_data = csvread('../fm50-2.csv');
0044 Data = file_data(:,1:end-1)';
0045 Labels = file_data(:, end)';
0046 Labels = Labels*2 - 1;
0047
0048 MaxIter = 20;
0049
0050
0051 TrainData = Data(:,1:2:end);
0052 TrainLabels = Labels(1:2:end);
0053
0054 ControlData = Data(:,2:2:end);
0055 ControlLabels = Labels(2:2:end);
0056
0057
0058 weak_learner = tree_node_w(3);
0059
0060
0061 [RLearners RWeights] = RealAdaBoost(weak_learner, TrainData, TrainLabels, MaxIter);
0062
0063
0064 ResultR = sign(Classify(RLearners, RWeights, ControlData));
0065
0066
0067 ErrorR = sum(ControlLabels ~= ResultR) / length(ControlLabels)
0068
0069 fid = fopen('DecisionRule.txt','w');
0070 code = TranslateToC(RLearners,RWeights,fid);
0071 fclose(fid);
0072
0073
0074
0075 tic
0076
0077 for i = 1: nFiles
0078
0079
0080 im = imread (strcat('MN/',iFiles(i).name));
0081 disp(['Processing im ' num2str(i) '...']);
0082
0083 im = medfilt2(im);
0084 imRGB = gray_to_rgb(im);
0085 [R, C] = size ( im );
0086
0087
0088
0089
0090
0091
0092
0093
0094
0095
0096
0097 for r = kh2+1:(R-kh2-1)
0098
0099 disp(['Processing row ' num2str(r) '...']);
0100
0101
0102 for c = kh2+1:(C-kh2-1)
0103
0104
0105 r0 = max ( 1, r-kh );
0106 c0 = max ( 1, c-kh );
0107 r1 = min ( R, r+kh );
0108 c1 = min ( C, c+kh );
0109
0110 im_sub = im ( r0:r1, c0:c1 );
0111
0112
0113
0114
0115 fv = get_feature_vector(im_sub,rm_k);
0116 fv_arr = [fv.i_m fv.i_std fv.rad_m];
0117
0118 conf_score = Classify(RLearners, RWeights, fv_arr');
0119 resultR = sign (conf_score);
0120 if ( resultR > 0 )
0121 imRGB(r,c,:) = [255 0 0];
0122 conf_map = [conf_map; r c conf_score];
0123 end
0124
0125 end
0126 end
0127
0128 frames = [frames; im2struct(imRGB)];
0129 imName_out = strcat('DT/',iFiles(i).name(1:end-4),'.dt.png');
0130 imwrite(imRGB,imName_out);
0131 end
0132
0133
0134
0135
0136
0137
0138
0139
0140
0141
0142 toc
0143 cd ..
0144 end