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:
|