CoolFace
Apppublic

aakash29786/demo

sourceHugging Facemitupdated 3y agoView on Hugging Face
0likes
App README

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:

  1. 1.app.py: Contains the primarily logic for uploading the images and inferecing on it using Resnet18.
  2. 2.resnet.py: Contains the resnet18 skeleton
  3. 3.downloaded.model.pth : is the trained (on Lightning) ResNet model that is saved and uploaded in app.py

Gradio Visualization: Input:

  1. 1.Input image upload
  2. 2.Transparency slider (0-100%)
  3. 3.Num of classes (1-10)
  4. 4.Which layer to target (-2,-1) for gradcam visualization

Output:

  1. 1.The top number of classes along with the image showing GradCam output