Skip to content

Image Classification, one of the techniques belonging to object recognition can be used in analysing objects appearing in an image.

Notifications You must be signed in to change notification settings

VidhiVijayvergiya/Image-Classification-using-CNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image-Classification-using-CNN

Image Classification, one of the techniques belonging to object recognition can be used in analysing objects appearing in an image.

With the combination of results of three CNNs trained on custom images , the project presents an application of image classification to analyze the behaviour of a dog whether it is calm, angry, playful, curious or want to grab your attention and the activity it is doing, be it sleeping, running or sitting.

The MODEL1 is the first CNN trained on the classes of angry and sleeping images of dog. The MODEL2 is the second CNN trained on the classes of running and sitting images of dog. The MODEL3 is the third CNN trained on the classes of open and closed mouth images of dog.

The results after training the three models can show how a proposed architecture of CNN performs on different classes.

The first step is to create a custom dataset and transform it to feed into the input layer. For the three CNNs, downloaded around 350 images from google belonging to each class and then used generic data augmentation techniques to increase the training dataset three times. Later, converted all the images to grayscale and a fixed size of 100x100 which sharpened all of them.

comparison

The following is the architecture of the CNN created, the information of the various layers and the activation function used:

Conv Architecture

We have trained three models which have given varying accuracy and this way we were able to observe and comment on how CNNs perform on different datasets.

Later we combined the output of all models and predicted the behaviour and activity done by the dog in the input image.

About

Image Classification, one of the techniques belonging to object recognition can be used in analysing objects appearing in an image.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published