Update visualization.py

visualization.py

@@ -235,11 +235,11 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, outpu
 
     # Custom colormap definition
     cdict = {
-        'red':   [(0.0,  0.0, 0.0),  # Low MSE: No red
+        'red':   [(0.0,  0.5, 0.5),  # Low MSE: 50% red (gray)
                   (1.0,  1.0, 1.0)], # High MSE: Full red
-        'green': [(0.0,  1.0, 1.0),  # Low MSE: Full green
+        'green': [(0.0,  0.5, 0.5),  # Low MSE: 50% green (gray)
                   (1.0,  0.0, 0.0)], # High MSE: No green
-        'blue':  [(0.0,  1.0, 1.0),  # Low MSE: Full blue
+        'blue':  [(0.0,  0.5, 0.5),  # Low MSE: 50% blue (gray)
                   (1.0,  0.0, 0.0)]  # High MSE: No blue
     }