vivek09thakur / ml-image-classification Goto Github PK

In this project, We are creating a machine learning classifier to classify and group similar kind of image to the sample images from the given dataset. The model leverages Keras-API to process the images and imagenet model to extract features of the images. So It can be classified based on the cosine similarity of sample image and rest of image in dataset.

• Dataset Used - Contains all the dataset including TrainData/ (contains all the images of dogs) and sampleImg/ (contains a single img of the germen sheperd)
• Training Data - Contains image of dogs of all breed (germen shepred,pug and golden retriver)
• SampleImg - Contains a single image of the handsome guy we used as a sample