How can I use a trained CNN to predict a new image label?
1 visualizzazione (ultimi 30 giorni)
Mostra commenti meno recenti
abich
il 16 Mag 2017
Commentato: Kristian Dubrie
il 10 Apr 2019
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 Commenti
Gergely Szépvölgyi
il 24 Set 2017
Has there been a solution to this since may? I'm facing the exact same problem here.
Risposta accettata
Cam Salzberger
il 16 Mag 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 Commenti
Più risposte (0)
Vedere anche
Categorie
Scopri di più su Image Data Workflows in Help Center e File Exchange
Prodotti
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!