Convolution Neural Network, Image Category Classification Using Deep Learning example
3 views (last 30 days)
Show older comments
Hello I have recently started working Convolution Neural Network. I found a tutorial https://au.mathworks.com/help/nnet/convolutional-neural-networks.html I have confusion in the "Image Category Classification Using Deep Learning" example. The example uses pre-trained model which is developed for the 1000 class. is it possible to use the model of 1000 class for 3 class?
1 Comment
Cy Pi
on 10 Jan 2017
Edited: Walter Roberson
on 10 Jan 2017
Yes, it's possible to use the model of 1000 class for 3 class. You just remove the last layer that has the 1000 classes and retrain it with 3 nodes at the last layer. There is a tool called Hive Moderation https://hivemoderation.com/ that does this.
Answers (1)
Walter Roberson
on 5 Jan 2017
No. The pretrained model already has the information about the other 997 classes built in, and you cannot just extract a subset of the information for 3 classes. If you wanted that you should extract the three classes from the original data and train with it.
You can use the pretrained model and apply it only to images intended to be in the three classes, but you cannot be sure it will not decide that the image belongs to one of the other classes.
2 Comments
Walter Roberson
on 5 Jan 2017
You need to train on your data, but the architecture might end up looking the same
See Also
Categories
Find more on Image Data Workflows in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!