We tried various deep-learning models for the dataset like the CNN model with different architectures, VGG 16.
Best model was the Vanilla CNN whose architecture can be seen below:
We chose AWS for our deployment services as it was a very convenient and flexible option.
Deployment Architecture :
Detailed Architecture:
Demo images:
We tried giving the below banana image for prediction.
Our prediction from the application :
But it was not performing well for some images like the below rotten orange image:
It was a prediction of Pomegranate Bad for Orange Bad.
We have to try various Deep Learning models for getting our predictions accurate.