Skip to content
Advertisement

Can CNN do better than pretrained CNN?

With all I know. pretrained CNN can do way better than CNN. I have a dataset of 855 images. I have applied CNN and got 94% accuracy.Then I applied Pretrained model (VGG16, ResNet50, Inception_V3, MobileNet)also with fine tuning but still i got highest 60% and two of them are doing very bad on classification. Can CNN really do better than pretrained model or my implementation is wrong. I’ve converted my image into 100 by 100 dimensions and followed the way of keras application. Then What is the issue ??

Naive CNN approach :

JavaScript

VGG16 approach:

JavaScript

Advertisement

Answer

What you’re referring to as CNN in both cases talk about the same thing, which is a type of a neural network model. It’s just that the pre-trained model has been trained on some other data instead of the dataset you’re working on and trying to classify.

What is usually used here is called Transfer Learning. Instead of freezing all the layers, trying leaving the last few layers open so they can be retrained with your own data, so that the pretrained model can edit its weights and biases to match your needs as well. It could be the case that the dataset you’re trying to classify is foreign to the pretrained models.

Here’s an example from my own work, there are additional pieces of code but you can make it work with your own code, the logic remains the same

JavaScript
User contributions licensed under: CC BY-SA
1 People found this is helpful
Advertisement