Update app.py
Browse files
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('
|
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 |
|