Spaces:
Sleeping
Sleeping
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" | |
![Image 1](cat.jpg) | |
![Image 2](dog.jpg) | |
![Image 3](bird_1.jpg) | |
![Image 4](cat_1.jpg) | |
![Image 5](cat_2.jpg) | |
![Image 6](dog_1.jpg) | |
![Image 7](dog_2.jpg) | |
![Image 8](dog_3.jpg) | |
![Image 9](ship_1.jpg) | |
![Image 10](ship_2.jpg) | |
app.py is the main application file | |