aakash29786/demo
0
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
GradCam output on CIFAR10 dataset with Resnet18
This code file demonstrates GradCam out on CIFAR10 dataset. It uses a trained ResNet18 model for inferecing of images.
Following are the code files:
- app.py: Contains the primarily logic for uploading the images and inferecing on it using Resnet18.
- resnet.py: Contains the resnet18 skeleton
- downloaded.model.pth : is the trained (on Lightning) ResNet model that is saved and uploaded in app.py
Gradio Visualization: Input:
- Input image upload
- Transparency slider (0-100%)
- Num of classes (1-10)
- Which layer to target (-2,-1) for gradcam visualization
Output:
- The top number of classes along with the image showing GradCam output
