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

7 views (last 30 days)
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 ?

Accepted Answer

Nagasai Bharat
Nagasai Bharat on 21 Apr 2021
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.

Sign in to comment.

More Answers (0)

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by