/Mobile-gallery-image-classifier

Distinguishes the mobile gallery's images into various sets such as whatsapp documents, nature, cars, screenshots, the machine learns itself and predicts the results with an accuracy of 97.6%

Primary LanguageJupyter Notebook

Mobile gallery image classifier

Everyone often finds their gallery to be messed up, with all the images mixed up together ,whether it’s any kind of information or any important documents,images or notes etc. This becomes a problem when ever you want any image then you have to look through the whole gallery and search out for that image and if you found it on time it’s your luck .This often poses a problem when anyone by chance delete any important images thinking it to be of waste. Everyone wants their images to be sorted and organized in such a way that whenever they search a particular image they can get it easily in fraction of time. In this project a Deep Learning (DL) based gallery image classification method is provided which can classify a given set of images into different categories such as trees, machines, memes etc. & hence make our phone gallery a better place to visit. The DL method used in this project is the Convolutional Neural Network (CNN). The dataset used for training consists of 6 categories –Whatsapp Screenshots, Memes, Selfies, Cars, Mountains and trees. The accuracy of the prediction is 97.6%.