Update visualization.py
Browse files- 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
|