awsaf49 commited on
Commit
094461a
·
1 Parent(s): 69a7cee
Files changed (1) hide show
  1. 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.4):
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: