update README.md

README.md

@@ -12,12 +12,17 @@ license: apache-2.0
 
 # IMAGE CLASSIFICATION with GradCAM
 
-A simple Gradio interface to visualize the output of a CNN trained on CIFAR10 dataset with GradCAM and Misclassified images. 
-The architecture is inspired from David Page’s (myrtle.ai) DAWNBench winning model archiecture. 
-Please input the image and select the number of top predictions to display - you will see the top predictions and their corresponding confidence scores. 
-You can also select whether to show GradCAM for the particular image (utilizes the gradients of the classification score with respect to the final convolutional feature map, to identify the parts of an input image that most impact the classification score). 
-You need to select the model layer where the gradients need to be plugged from - this affects how much of the image is used to compute the GradCAM. You can also select whether to show misclassified images - these are the images that the model misclassified. 
-Some examples are provided in the examples tab.
+A simple Gradio interface to visualize the output of a CNN trained on **CIFAR10 dataset** with **GradCAM** and **Misclassified images**.
+
+The architecture is inspired from David Page’s (myrtle.ai) DAWNBench winning model archiecture. Please refer to https://myrtle.ai/learn/how-to-train-your-resnet-8-bag-of-tricks/ to know more.
+
+
+### Instructions
+1. Please input the image and select the number of top predictions to display - you will see the top predictions and their corresponding confidence scores.
+2. You can also select whether to show GradCAM for the particular image (utilizes the gradients of the classification score with respect to the final convolutional feature map, to identify the parts of an input image that most impact the classification score). 
+3. You need to select the model layer where the gradients need to be plugged from - this affects how much of the image is used to compute the GradCAM.
+4. You can also select whether to show misclassified images - these are the images that the model misclassified. Please select the number of misclassified images to display - the pipeline selects the bunch randomly from the misclassified images in the test set.
+5. Some examples are provided in the examples tab.
 
 ## CIFAR-10 Dataset
 
@@ -25,5 +30,22 @@ The CIFAR-10 dataset consists of 60000 32x32 colour images in 10 classes, with 6
 
 The dataset is divided into five training batches and one test batch, each with 10000 images. The test batch contains exactly 1000 randomly-selected images from each class. The training batches contain the remaining images in random order, but some training batches may contain more images from one class than another. Between them, the training batches contain exactly 5000 images from each class.
 
+![Data Samples](Store/image.png)
+
+## Model Metrics
+| Train Acc | Test Acc | Train Loss | Test Loss |
+|-----------|----------|------------|-----------|
+| 96.47     | 92.50    | 0.10       | 0.23      |
+
+![image](https://github.com/Madhur-1/ERA-v1/assets/64495917/99f9bb9d-d907-41f5-b134-a214750b1c4b)
+
+
+## Grad-CAM
+Note: The following has been taken from https://towardsdatascience.com/understand-your-algorithm-with-grad-cam-d3b62fce353
+
+Gradient-weighted Class Activation Mapping (Grad-CAM), uses the gradients of any target concept (say ‘dog’ in a classification network or a sequence of words in captioning network) flowing into the final convolutional layer to produce a coarse localization map highlighting the important regions in the image for predicting the concept.
+
+![Alt text](Store/image-2.png)
 
+------
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference