drhead commited on
Commit
06cf327
·
verified ·
1 Parent(s): db94ebe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -248,7 +248,7 @@ def create_cam_visualization_pil(image_pil, cam, alpha=0.6, vis_threshold=0.2):
248
  cam_norm = (cam - cam.min()) / (np.ptp(cam) + 1e-8)
249
 
250
  # Create heatmap using matplotlib colormap
251
- colormap = cm.get_cmap('jet')
252
  cam_colored = colormap(cam_norm)[:, :, :3] # RGB
253
  cam_alpha = (cam_norm >= vis_threshold).astype(np.float32) * alpha # Alpha mask
254
 
 
248
  cam_norm = (cam - cam.min()) / (np.ptp(cam) + 1e-8)
249
 
250
  # Create heatmap using matplotlib colormap
251
+ colormap = cm.get_cmap('inferno')
252
  cam_colored = colormap(cam_norm)[:, :, :3] # RGB
253
  cam_alpha = (cam_norm >= vis_threshold).astype(np.float32) * alpha # Alpha mask
254