Image-classification-using-DNN-and-CNN 🔍

Aim 🧩

The main aim of this project is to understand and implement deep neural network (DNN) and convolutional neural network (CNN) and see how these can be used for classification problems.

Built using 🛠️

• Python

Dataset used is 📊

In this we will use a subset of images from Tiny ImageNet, which is a subset of the ImageNet dataset. Our subset of Tiny ImageNet contains 30 different categories, we will refer to it as TinyImageNet30. The training set has 450 resized images (64x64 pixels) for each category (13,500 images in total).

Required Packages 📦

• numpy
• h5py
• matplotlib
• pytorch

How the work has been divided 👩🏻‍💻

1. Function Implementation - to get the datasets and the dataloaders
2. Function Implementation - model class for a simple MLP and CNN Model
3. Model Training
4. Generating Confusion Matrices and ROC
5. Implementing strategies to tackle overfitting
6. Model Testing
7. Model Fine-tuning on the CIFAR-10 dataset