How to choose a pre-trained neural network for your image classification

Question

Abdulaziz Alotaibi 2021-4-18

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/805936-how-to-choose-a-pre-trained-neural-network-for-your-image-classification

编辑： Nagasai Bharat 2021-4-22

采纳的回答： Nagasai Bharat

Hello,

I'm looking into the pre-trained neural networks such as : mobilenet , googlenet, resnet 50, VGG16 and alex net

and I'm wondering how to choose one of them to transfer learning?

my image classification program is simple and it's about detecting traffic signs, but the question is how should i choose one of the above networks?

how do I know which one of them will perform better than the others ?

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

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

Answer 1

Nagasai Bharat 2021-4-21

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/805936-how-to-choose-a-pre-trained-neural-network-for-your-image-classification#answer_680745

Hi,

From my understanding you are trying to use a pretrained model for an image classification task. In order to do that you can find the available pretrained networks from this doc. The doc provides a section on Transfer learning . And an example on how to use this network can be accessed from here. The example also shows how to evaluate the network too.

3 个评论
显示 1更早的评论隐藏 1更早的评论

Nagasai Bharat 2021-4-22

编辑：Nagasai Bharat 2021-4-22

The choice would be on the requirement of the project you are working on. The time the model takes to predict, memory contraint of the model parameters, the harware you are using to train(more layers implies more time taken by the learning algorithm) and finally the accuracy you are getting without overfiting.

An option would be to try each model and cross-check their acceracy metric and choose the best one also taking into consideration of the above requirements.

Machine Learning Challenges: Choosing the Best Classification Model and Avoiding Overfitting - MATLAB & Simulink (mathworks.com) - This could help you

Abdulaziz Alotaibi 2021-4-22