reab5555 commited on
Commit
c530669
·
verified ·
1 Parent(s): a588df0

Update visualization.py

Browse files
Files changed (1) hide show
  1. visualization.py +1 -0
visualization.py CHANGED
@@ -314,6 +314,7 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
314
  print(f"Failed to create heatmap video at: {heatmap_video_path}")
315
  return None
316
 
 
317
  # Function to create the correlation heatmap
318
  def plot_correlation_heatmap(mse_embeddings, mse_posture, mse_voice):
319
  data = np.vstack((mse_embeddings, mse_posture, mse_voice)).T
 
314
  print(f"Failed to create heatmap video at: {heatmap_video_path}")
315
  return None
316
 
317
+
318
  # Function to create the correlation heatmap
319
  def plot_correlation_heatmap(mse_embeddings, mse_posture, mse_voice):
320
  data = np.vstack((mse_embeddings, mse_posture, mse_voice)).T