Spaces:
Sleeping
Sleeping
Commit
·
6373a51
1
Parent(s):
30b6d93
Added examples
Browse files
app.py
CHANGED
|
@@ -39,7 +39,7 @@ def resize_image_pil(image, new_width, new_height):
|
|
| 39 |
|
| 40 |
return resized
|
| 41 |
|
| 42 |
-
def inference(input_img, transparency = 0.5,
|
| 43 |
input_img = resize_image_pil(input_img, 32, 32)
|
| 44 |
|
| 45 |
input_img = np.array(input_img)
|
|
@@ -73,13 +73,22 @@ def inference(input_img, transparency = 0.5, is_grad_cam=True, target_layer_numb
|
|
| 73 |
|
| 74 |
title = "CIFAR10 trained on ResNet18 Model with GradCAM"
|
| 75 |
description = "A simple Gradio interface to infer on ResNet model, and get GradCAM results"
|
| 76 |
-
examples = [["cat.jpg", 0.5, -1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 77 |
demo = gr.Interface(
|
| 78 |
inference,
|
| 79 |
inputs = [
|
| 80 |
gr.Image(width=256, height=256, label="Input Image"),
|
| 81 |
-
gr.Slider(0, 1, value = 0.5, label="Overall Opacity of Image"),
|
| 82 |
gr.Checkbox(label="Show GradCAM"),
|
|
|
|
| 83 |
gr.Slider(-2, -1, value = -2, step=1, label="Which Layer?"),
|
| 84 |
gr.Slider(2, 10, value=3, step=1, label="Number of Top Classes")
|
| 85 |
],
|
|
|
|
| 39 |
|
| 40 |
return resized
|
| 41 |
|
| 42 |
+
def inference(input_img, is_grad_cam=True, transparency = 0.5, target_layer_number = -1, top_predictions=3):
|
| 43 |
input_img = resize_image_pil(input_img, 32, 32)
|
| 44 |
|
| 45 |
input_img = np.array(input_img)
|
|
|
|
| 73 |
|
| 74 |
title = "CIFAR10 trained on ResNet18 Model with GradCAM"
|
| 75 |
description = "A simple Gradio interface to infer on ResNet model, and get GradCAM results"
|
| 76 |
+
examples = [["cat.jpg", True, 0.5, -1, 3],
|
| 77 |
+
["dog.jpg", True, 0.5, -1, 3],
|
| 78 |
+
["bird.jpg", True, 0.5, -1, 3],
|
| 79 |
+
["car.jpg", True, 0.5, -1, 3],
|
| 80 |
+
["deer.jpg", True, 0.5, -1, 3],
|
| 81 |
+
["frog.jpg", True, 0.5, -1, 3],
|
| 82 |
+
["horse.jpg", True, 0.5, -1, 3],
|
| 83 |
+
["plane.jpg", True, 0.5, -1, 3],
|
| 84 |
+
["ship.jpg", True, 0.5, -1, 3],
|
| 85 |
+
["truck.jpg", True, 0.5, -1, 3]]
|
| 86 |
demo = gr.Interface(
|
| 87 |
inference,
|
| 88 |
inputs = [
|
| 89 |
gr.Image(width=256, height=256, label="Input Image"),
|
|
|
|
| 90 |
gr.Checkbox(label="Show GradCAM"),
|
| 91 |
+
gr.Slider(0, 1, value = 0.5, label="Overall Opacity of Image"),
|
| 92 |
gr.Slider(-2, -1, value = -2, step=1, label="Which Layer?"),
|
| 93 |
gr.Slider(2, 10, value=3, step=1, label="Number of Top Classes")
|
| 94 |
],
|
bird.jpg
ADDED
|
car.jpg
ADDED
|
deer.jpg
ADDED
|
frog.jpg
ADDED
|
horse.jpg
ADDED
|
plane.jpg
ADDED
|
ship.jpg
ADDED
|
truck.jpg
ADDED
|