Spaces:
Build error
Build error
Commit
·
cd6a639
1
Parent(s):
f2dcc55
updated to show only predicted value for user selected images
Browse files
app.py
CHANGED
|
@@ -188,8 +188,10 @@ def process_gradcam_images(num_images,layer,opacity,image_list=None):
|
|
| 188 |
rgb_img = rgb_img.numpy()
|
| 189 |
|
| 190 |
visualization = show_cam_on_image(rgb_img, grayscale_cam, use_rgb=True, image_weight=opacity)
|
| 191 |
-
|
| 192 |
-
|
|
|
|
|
|
|
| 193 |
|
| 194 |
print(str(num_images) + "**" + str(layer) + "**" + str(opacity))
|
| 195 |
return grad_images, gr.update(visible=True)
|
|
|
|
| 188 |
rgb_img = rgb_img.numpy()
|
| 189 |
|
| 190 |
visualization = show_cam_on_image(rgb_img, grayscale_cam, use_rgb=True, image_weight=opacity)
|
| 191 |
+
if not image_list:
|
| 192 |
+
grad_images.append(((visualization),f'Pred: {classes[pred.cpu()]} | Truth :{classes[correct.cpu()]}'))
|
| 193 |
+
else:
|
| 194 |
+
grad_images.append(((visualization),f'Prediction: {classes[pred.cpu()]}'))
|
| 195 |
|
| 196 |
print(str(num_images) + "**" + str(layer) + "**" + str(opacity))
|
| 197 |
return grad_images, gr.update(visible=True)
|