Convolution Neural Network, Image Category Classification Using Deep Learning example

Question

Adnan Farooq Awan 2017-1-5

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/319130-convolution-neural-network-image-category-classification-using-deep-learning-example

编辑： Walter Roberson 2017-1-10

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 个评论
显示 -1更早的评论隐藏 -1更早的评论

Cy Pi 2017-1-10

编辑：Walter Roberson 2017-1-10

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.

请先登录，再进行评论。

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

Answer 1

Walter Roberson 2017-1-5

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/319130-convolution-neural-network-image-category-classification-using-deep-learning-example#answer_249450

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.