Improve performance of my CNN model

I am working on an image classification problem. There are 876 images in the training and 600 in the test dataset. It is a multi class classification for plants.

Since this is my first CNN problem, I started working with tensorflow and keras to build my model and then started using transfer learning to improve my performance. The best model which I built so far using Keras is getting me a score of 0.68 ( since the test dataset has no labels, I have to upload my results over and they generate the score using log-loss).

Also, I tried using Fastai and Pytorch which significantly improved my single model score to 0.46.

I have used Resnet50 image dataset for both cases but i cannot experiment using Resnet101 or increasing the batch size since my gpu goes out of memory( also tried using kaggle and google colab).

Is their any way which i can try to further improve the performance? I am basically stuck now since i can't experiment by increasing either batch size or using a better model like Resnet101 etc.

I have uploaded both codes at : https://github.com/akshatshreemali/ImageClassification

Topic fastai keras image-classification performance

Category Data Science

About

Geeks Mental is a community that publishes articles and tutorials about Web, Android, Data Science, new techniques and Linux security.