基于MATLAB编程极限学习机ELM的草莓等级识别，ELM图像分类

clc clear close all load maydata2.mat aa = [ones(length(num1),1);2*ones(length(num2),1);3*ones(length(num3),1)]; num = [num1;num2;num3]; nn = length(num); bb = zeros(nn,3); for ii = 1:nn bb(ii,aa(ii))=1; end % n = randperm(nn); load maydatan.mat input_train=num(n(1:round(0.8*nn)),1:4)';%训练数据的输入数据 output_train=bb(n(1:round(0.8*nn)),:)';%训练数据的输出数据 input_test=num(n(round(0.7*nn+1:nn)),1:4)';%测试数据的输入数据 output_test=aa(n(round(0.7*nn+1:nn)),1)'; %测试数据的输出数据 %% 归一化 % 训练集 [Pn_train,inputps] = mapminmax(input_train,0,1); [Tn_train,outputps] = mapminmax(output_train,0,1); % 测试集 Pn_test = mapminmax('apply',input_test,inputps); % Tn_test = mapminmax('apply',output_test,outputps); % tic %% ELM创建/训练 [IW,B,LW,TF,TYPE] = elmtrain(Pn_train,Tn_train,100,'sig',0); %% ELM仿真测试 Tn_sim = elmpredict(Pn_test,IW,B,LW,TF,TYPE);%测试 Tn_simp1 = elmpredict(Pn_train,IW,B,LW,TF,TYPE);%训练 % 反归一化 [mx,T_sim] = max(mapminmax('reverse',Tn_sim,outputps)); T_simp = mapminmax('reverse',Tn_simp1,outputps); % toc %% 结果对比 result = [output_test T_sim]; % 均方误差 % E = mse(T_sim - T_test); % 决定系数 % N = length(T_test); % R2 = (N*sum(T_sim.*T_test)-sum(T_sim)*sum(T_test))^2/((N*sum((T_sim).^2)-(sum(T_sim))^2)*(N*sum((T_test).^2)-(sum(T_test))^2)); %% 绘图 figure plot(output_test(1,:),'r-*') hold on plot(T_sim(1,:),'b-o') hold off ylabel('成熟级别') title('50949') legend('期望输出','预测输出') error = output_test-T_sim; jingdu = (sum(output_test==T_sim))/length(T_sim); figure plot(error,'k*') ylabel('误差') title('ELM测试集预测误差') title('50949') [fileName,pathName] = uigetfile('*.*','Please select an image');%文件筐，选择文件 if(fileName) fileName = strcat(pathName,fileName); fileName = lower(fileName);%一致的小写字母形式 else msgbox('Please select an image'); return; %退出程序 end imgb = imread(fileName);% 读取原始图像; [mx1,mx2,mx3] = size(imgb); numb = [reshape(imgb(:,:,1),mx1*mx2,1) reshape(imgb(:,:,2),mx1*mx2,1) reshape(imgb(:,:,3),mx1*mx2,1)]; load maynet2.mat [ rgb1 ] =fenge(imgb,numb,net,inputps,mx1,mx2); [Pn_train,inputps] = mapminmax(input_train,0,1); % 测试集 Pn_testb = mapminmax('apply',rgb1',inputps); Tn_simb = elmpredict(Pn_testb,IW,B,LW,TF,TYPE);%测试 [mx,T_simb] = max(mapminmax('reverse',Tn_simb,outputps)); str1 = {'特级','一级','二级'}; figure imshow(imgb) text(1500.5,-60.1,str1{T_simb},'Color','red','FontSize',24)