How can I use a trained CNN to predict a new image label?
1 view (last 30 days)
Show older comments
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?
2 Comments
Gergely Szépvölgyi
on 24 Sep 2017
Has there been a solution to this since may? I'm facing the exact same problem here.
Accepted Answer
Cam Salzberger
on 16 May 2017
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 Comments
More Answers (0)
See Also
Categories
Find more on Image Data Workflows in Help Center and File Exchange
Products
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!