Update visualization.py

visualization.py

@@ -228,9 +228,9 @@ def create_heatmap(t, mse_embeddings, mse_posture, mse_voice, video_fps, total_f
     mse_voice_norm = (mse_voice - np.min(mse_voice)) / (np.max(mse_voice) - np.min(mse_voice))
     
     combined_mse = np.zeros((3, total_frames))
-    combined_mse[0] = mse_embeddings_norm
-    combined_mse[1] = mse_posture_norm
-    combined_mse[2] = mse_voice_norm
+    combined_mse[0] = mse_embeddings
+    combined_mse[1] = mse_posture
+    combined_mse[2] = mse_voice
 
     fig, ax = plt.subplots(figsize=(video_width / 300, 0.4))
     ax.imshow(combined_mse, aspect='auto', cmap='Reds', vmin=0, vmax=1, extent=[0, total_frames, 0, 3])