Im trying to create an image classification model that classifies plants from an image dataset made up of 33 classes, the total amount of images is 41,808, the images are unbalanced but that is something me and my thesis team will work on using Kfold; but going back to the main problem.

The VGG16 model itself is from a pre-trained model from keras

My source code should be attached in this question (paste_1292099)

The results of a 15-epoch run is also attached as well

what I have done so far is changing the optimizers from SGD to Adam, but the results are generally the same.

Am I doing something wrong or is there anything I can do to improve on this model to get it to atleast be in a "working" state, regardless if its overfitting or the like as that can be fixed later.

This is also the link to our dataset:

https://drive.google.com/file/d/134CTAf3KNFrm9Lm2IUWQn1GjCG1UeD_s/view?usp=sharing

It is specifically a dataset consisting of Medicinal Plants and Herbs in our region with their augmentations. The are not yet resized and normalized in the dataset.

Similar questions and discussions