reab5555 commited on
Commit
a882402
·
verified ·
1 Parent(s): 51611cc

Update visualization.py

Browse files
Files changed (1) hide show
  1. visualization.py +1 -1
visualization.py CHANGED
@@ -201,7 +201,7 @@ def plot_posture(df, posture_scores, color='blue', anomaly_threshold=3):
201
  return fig
202
 
203
 
204
- def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_voice, output_path):
205
  from matplotlib.backends.backend_agg import FigureCanvasAgg
206
  # Open the video
207
  cap = cv2.VideoCapture(video_path)
 
201
  return fig
202
 
203
 
204
+ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, output_path):
205
  from matplotlib.backends.backend_agg import FigureCanvasAgg
206
  # Open the video
207
  cap = cv2.VideoCapture(video_path)