Data Science Stack Exchange is a question and answer site for Data science professionals, Machine Learning specialists, and those interested in learning more about the field. It only takes a minute to sign up.
Sign up to join this community
Is there a golden rule which gives intuition on which base model needs to be used for a give image classification problem.
Most of the articles gives the below details which says how to train the model based on the dataset.
However I was not able to find good reference for the selection of base model.
Thank you
There is no specific rule associated with the base model selection for transfer learning. It is generally a trade-off between model precision and resource allocation. As the number of layers increases, the number of parameters increases and the model becomes more and more resource heavy but deeper models tend to have better accuracy over shallower counterparts. Here's a comparison:
Apart from that also refer to this question: Which is the fastest image pre-trained model? if the speed of the model matters.
