updated
Browse files- grad_cam.py +1 -1
grad_cam.py
CHANGED
@@ -6,7 +6,7 @@ import numpy as np
|
|
6 |
import matplotlib.cm as cm
|
7 |
|
8 |
class GradCam:
|
9 |
-
def __init__(self, model, img,
|
10 |
self.model = model
|
11 |
self.img_path = img
|
12 |
self.size = size
|
|
|
6 |
import matplotlib.cm as cm
|
7 |
|
8 |
class GradCam:
|
9 |
+
def __init__(self, model, img, last_conv_layer_name, pred_index=None):
|
10 |
self.model = model
|
11 |
self.img_path = img
|
12 |
self.size = size
|