How can I use a trained CNN to predict a new image label?

1 次查看(过去 30 天)
I was applying the following CNN fine-tuning example : https://www.mathworks.com/help/nnet/examples/transfer-learning-and-fine-tuning-of-convolutional-neural-networks.html
The example shows how to fine-tune a pre-trained CNN on letters to classify images of digits. Now i would like to use this new fine-tuned CNN on new images of digits that i have on my computer. How can I do that?

采纳的回答

Cam Salzberger
Cam Salzberger 2017-5-16
Hello Abich,
The example you linked shows how to train on a new set of images. I am not a neural networks expert, but if the output you are looking for is the same as the output from the retrained network (digits 0-9), I do not even think you need to replace the final three layers of the network. You can just create the image datastore for your new images, set any options you would like with "trainingOptions", and then call "trainNetwork".
Did you have some specific questions about a particular step?
-Cam
  7 个评论
An Grg
An Grg 2018-4-16
编辑:An Grg 2018-4-16
I just wanted to know did it work the classifier? I am having a problem trying to output the probability/predict the testing image.
Thanks and would appreciate reply.

请先登录,再进行评论。

更多回答(0 个)

类别

Help CenterFile Exchange 中查找有关 Image Data Workflows 的更多信息

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by