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
|