Neural Network Code for Image Classification/Segmentation

1 次查看（过去 30 天）

Anis Mahmon 2015-8-28

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/237177-neural-network-code-for-image-classification-segmentation

评论： Greg Heath 2015-8-28

I have written a code in Matlab for neural network classification/segementation the image after train the image. But, it is not working as expected. I get the error in this line "yhat = net(double([r;g;b]));". Is there any mistake in the code? Can you take a look at the code and help me out !!

a = imread('C:\Users\user\Desktop\MATLAB PROCESSING\Matlab Processing\Band432\open land\klang.tif');
b = rgb2gray(a);
figure, imshow(a);
figure, imshow(b);
thr = 0.5;
for i = 1:1:size(a,1)
  for j = 1:1:size(a,2)
      pix = a(i,j,:);
      r = pix(1,1,1);
      g = pix(1,1,2);
      b = pix(1,1,3);
      yhat = net(double([r;g;b]));
      if yhat >= thr
          result = 1;
          b(i,j) = 255;
      else
          result = 0;
          b(i,j) = 0;
      end
  end
end
figure3, imshow(b)