Don’t lose your preferences!

Sign-in or sign-up to save your personalization preferences to return to later
FeedChevron right imageData ScienceChevron right imageNeural Network for Image ClassificationChevron right image
Post main image

Neural Network for Image Classification

Extracting features on images using deep learning models

Badge image#2 Project of the Week

The project goal is an image classification problem for deep learning models. So I explored a simple neural network, and then progressed to convolutional neural network and transfer learning. First I started with image classification using a Simple Neural Network. The dataset is from pyimagesearch, which has 3 classes: cat, dog, and panda. There are 3000 images in total, ie, 1000 for each class. Then I studied the Convolutional Neural Network, where each image goes through a series of convolution and max pooling for features extraction. I explored using the CIFAR-10 dataset which has 60,000 images divided into 10 classes. Next I explored a huge dataset of over a million images, using Transfer Learning to avoid reinventing the wheel. I used the VGG16 pre-trained model developed by University of Oxford, which has 1000 classes ranging from animals to things and food. The conclusion is: Image classification can be done using neural network models. Identifying patterns and extracting features on images are what deep learning models can do, and they do it very well. More details on this project: Medium: Github:

Video demo
Project screenshot 1
Project screenshot 2
Project screenshot 3
Project screenshot 4
Project screenshot 5
Comment post

Skip straight to final coding interviews with
Remote Companies!

Top Makers

Most active posts

Top projects