wrong classification caffe google net model in matlab

3 次查看（过去 30 天）

nitin kandpal 2019-10-6

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/483800-wrong-classification-caffe-google-net-model-in-matlab

评论： Piyush Dubey 2019-10-10

Hi all,

I trained a google net model for two claases in Nvidia DIGITS which is working well in in DIGITS6.0 with 99.7 accuracy in test set.

Same model imported in matlab not able to classify properly. I am thinking it has to do something with mean image or binary.proto file.

I am not getting any error just wrong classification and i am sure my network is woking properly in DIGITS6.0 and python.

below is the matlab code

protofile = 'deploy.prototxt';

datafile = 'snapshot_iter_2295.caffemodel';

net = importCaffeNetwork(protofile,datafile);

inputSize = net.Layers(1).InputSize;

% image path

img = 'Test/Normal/0.JPEG';

im1 = imread(img);

im2 = rgb2gray(im1);

im = imresize(im2,inputSize(1:2)) ;

[label,score] = classify(net,im); % Classify the picture

Thanks

Nitin

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Piyush Dubey 2019-10-10

在 MATLAB Online 中打开

Hi Nitin,

When you obtain 'im2' by using 'rgb2gray' function, the image size changes from [224 224 3] to [224 224]. When you try to classify it, you should get an error saying 'Incorrect input size. The input images must have a size of [224 224 3].'. Can you try classifying after removing the following line from your code:

im2 = rgb2gray(im1);

And changing the line where you resize the image im2 to the following:

im = imresize(im1, inputSize(1:2));

Also, see if the following link can help:

https://www.mathworks.com/help/deeplearning/ref/importcaffenetwork.html#mw_819279d4-efd0-482b-9595-0153894143ad

请先登录，再进行评论。

请先登录，再回答此问题。