Dataset Link - https://www.kaggle.com/lsind18/gemstones-images
This dataset contains 3,200+ images of different gemstones. The images are grouped into 87 classes which are already divided into train and test data. All images are in various sizes and are in .jpeg format.
The objective is to predict to which class a particular gemstone belongs.
- Importing our data.
- Data Augmentation.
- Model Building.
- Flattening and Adding Dense Layers.
- Using vgg19 Tranfer learning techniques.
- Passing it into the gradio app
- Concluding our analysis by testing the model with some random user input.
The Model with the help of gradio is able to make a decent prediction in which class a particular gemstone belongs to.
