README.md

metadata

title: Erav2s13demo
emoji: 👀
colorFrom: green
colorTo: indigo
sdk: gradio
sdk_version: 4.29.0
app_file: app.py
pinned: false
license: mit

Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

This project is regarding ERA V2 Assignment 13 where we have used CIFAR10 Dataset and further applied GRAD-CAM on it

cifar10_saved_model.pth - is the saved model file

resnet.py - is the resnet 18 model architecture. In this we have used Pytorch LightningModule to faster up our coding implementation. Total parameter count is 11.2 M

The requirements for this project is given below. We have mentioned the required libraries in requirements.txt file

1. pytorch-lightning
2. torch
3. torchvision
4. torch-lr-finder
5. grad-cam
6. pillow
7. numpy

Our main aim of the project was to

1. ask the user whether he/she wants to see GradCAM images and how many, and from which layer, allow opacity change as well
2. ask whether he/she wants to view misclassified images, and how many
3. allow users to upload new images, as well as provide 10 example images
4. ask how many top classes are to be shown (make sure the user cannot enter more than 10)

We have uploaded some sample images to test our application. The sample images are "cat.jpg", "dog.jpg", "bird_1.jpg", "cat_1.jpg", "cat_2.jpg", "dog_1.jpg", "dog_2.jpg", "dog_3.jpg","ship_1.jpg","ship_2.jpg"

app.py is the main application file

Below is the validation accuracy screenshot. Our maximum validation accuracy is 71%

Below is the tensorboard logs