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