Update visualization.py

visualization.py

@@ -230,7 +230,7 @@ def create_heatmap(t, mse_embeddings, mse_posture, mse_voice, video_fps, total_f
     combined_mse[1] = mse_posture_norm
     combined_mse[2] = mse_voice_norm
 
-    fig, ax = plt.subplots(figsize=(video_width / 240, 0.6))
+    fig, ax = plt.subplots(figsize=(video_width / 240, 0.5))
     ax.imshow(combined_mse, aspect='auto', cmap='Reds', vmin=0, vmax=1, extent=[0, total_frames, 0, 3])
     ax.set_yticks([0.5, 1.5, 2.5])
     ax.set_yticklabels(['Voice', 'Posture', 'Face'], fontsize=7) 
@@ -264,10 +264,10 @@ 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)
     
-    # Fill missing MSE values with 0
+    # Fill missing MSE values with 0.001
     def pad_with_zeros(mse_array, total_frames):
         if len(mse_array) < total_frames:
-            return np.pad(mse_array, (0, total_frames - len(mse_array)), 'constant', constant_values=0)
+            return np.pad(mse_array, (0, total_frames - len(mse_array)), 'constant', constant_values=0.001)
         else:
             return mse_array[:total_frames]