Update visualization.py

visualization.py

@@ -203,13 +203,6 @@ def plot_posture(df, posture_scores, color='blue', anomaly_threshold=3):
     plt.close()
     return fig
 
-
-import matplotlib.pyplot as plt
-from matplotlib.backends.backend_agg import FigureCanvasAgg
-import matplotlib.colors as mcolors
-import numpy as np
-import cv2
-
 def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, output_path, desired_fps):
     cap = cv2.VideoCapture(video_path)
     original_fps = cap.get(cv2.CAP_PROP_FPS)
@@ -242,7 +235,7 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, outpu
     fig, ax = plt.subplots(figsize=(width/100, 2))
     im = ax.imshow(combined_mse, aspect='auto', extent=[0, total_frames, 0, 2])
     ax.set_yticks([0.5, 1.5])
-    ax.set_yticklabels(['Facial', 'Posture'])
+    ax.set_yticklabels(['Face', 'Posture'])
     ax.set_xticks([])
     plt.tight_layout()