Update visualization.py

visualization.py

@@ -264,13 +264,16 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
     width, height = video.w, video.h
     total_frames = int(video.duration * video.fps)
     
+    def fill_with_zeros_and_values(mse_array, total_frames):
+        result = np.zeros(total_frames)
+        indices = np.linspace(0, total_frames - 1, len(mse_array)).astype(int)
+        result[indices] = mse_array
+        return result
+
     # Ensure all MSE arrays have the same length as total_frames
-    mse_embeddings = np.interp(np.linspace(0, len(mse_embeddings) - 1, total_frames), 
-                               np.arange(len(mse_embeddings)), mse_embeddings)
-    mse_posture = np.interp(np.linspace(0, len(mse_posture) - 1, total_frames), 
-                            np.arange(len(mse_posture)), mse_posture)
-    mse_voice = np.interp(np.linspace(0, len(mse_voice) - 1, total_frames), 
-                          np.arange(len(mse_voice)), mse_voice)
+    mse_embeddings = fill_with_zeros_and_values(mse_embeddings, total_frames)
+    mse_posture = fill_with_zeros_and_values(mse_posture, total_frames)
+    mse_voice = fill_with_zeros_and_values(mse_voice, total_frames)
     
     def combine_video_and_heatmap(t):
         video_frame = video.get_frame(t)