alpha=0.6
Browse files- gcvit/utils/gradcam.py +1 -1
gcvit/utils/gradcam.py
CHANGED
@@ -17,7 +17,7 @@ def get_gradcam_model(model):
|
|
17 |
preds = model.forward_head(feats)
|
18 |
return tf.keras.models.Model(inp, [preds, feats])
|
19 |
|
20 |
-
def get_gradcam_prediction(img, grad_model, process=True, decode=True, pred_index=None, cmap='jet', alpha=0.
|
21 |
"""Grad-CAM for a single image
|
22 |
|
23 |
Args:
|
|
|
17 |
preds = model.forward_head(feats)
|
18 |
return tf.keras.models.Model(inp, [preds, feats])
|
19 |
|
20 |
+
def get_gradcam_prediction(img, grad_model, process=True, decode=True, pred_index=None, cmap='jet', alpha=0.6):
|
21 |
"""Grad-CAM for a single image
|
22 |
|
23 |
Args:
|