Solved – How to use deep learning (Convolution Neural Network) with small training data-set

conv-neural-networkdeep learningtrain

I have an image data set of around 180 images in 60 classes (3 images per class). I am able to build a classifier using feature matching. However, I want to try Convolution Neural Networks and see if I can improve.

I don't think I can train a CNN on just 120 images.

If that's true, what is the best way to approach this problem?

Best Answer

The way I solved the problem was that I took a neural network trained on imagenet and then extracted the features from one of the last layers and applied SVM on the feature vectors from these layers.

These gave me 75% accuracy on the dataset.