Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def inference(input_image, transparency=0.5, target_layer_number=-1):
|
|
55 |
_, prediction = torch.max(outputs, 1)
|
56 |
target_layers = [model.layer2[target_layer_number]]
|
57 |
cam = GradCAM(model= model, target_layers = target_layers)
|
58 |
-
grayscale_cam = cam(input_tensor=input_image,
|
59 |
grayscale_cam = grayscale_cam[0, :]
|
60 |
visualization = show_cam_on_image(
|
61 |
org_img/255,
|
|
|
55 |
_, prediction = torch.max(outputs, 1)
|
56 |
target_layers = [model.layer2[target_layer_number]]
|
57 |
cam = GradCAM(model= model, target_layers = target_layers)
|
58 |
+
grayscale_cam = cam(input_tensor=input_image, targets=None)
|
59 |
grayscale_cam = grayscale_cam[0, :]
|
60 |
visualization = show_cam_on_image(
|
61 |
org_img/255,
|