0
$\begingroup$

I have a question about the possible outcome of a trained model. Imagine that I would like to classify 2 different models of Ferrari and the dataset of these 2 models is small (for example, a few hundred images per model).

In the Keras blog, the issue was discussed but in the example, they are classifying dogs and cats, 2 class that are very general, distinct between them, and there are many cats and dogs already included in the original imagenet model (lets say the car/bus/truck are not included in the imagenet output classes).

Will it be better start training the model from Coco/Imagenet weights or start training using the weights of a model previously trained to classify cars only?

$\endgroup$
0
$\begingroup$

If you have a previously trained model to classify only cars, you can use it considering the fact that you have less data. The trained model for cars classification that you have already, would have been fine-tuned on the Imagenet or COCO in the first place. This way I think you would be able to train the model in less amount of time compared to fine tuning on Imagenet weights. Try both the methods once and try to find out the amount of performance increase in terms of time and accuracy over each other.

| improve this answer | |
$\endgroup$
0
$\begingroup$

I would rather suggest you to create your own CNN for this task and train it from scratch and then tune it's hyperparameters and i think that will give you much more better results as compared to using a pretrained model, these models have deep architectures and have millions of parameters which are learnt over a huge dataset, and these models are deep and they were made solely because of the fact that training such a deep network with so many parameters will require a good amount of data because i don't think so there any ferrari's in the coco dataset so doing training on top of that won't increase your model power upto what extent you may be wanting because you don't have that much data.So, why not build a CNN from your own hands your own network that should give you better results because as you said you have only two classes to classify.

| improve this answer | |
$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.