reab5555 commited on
Commit
501c305
·
verified ·
1 Parent(s): 1db5ddf

Update video_processing.py

Browse files
Files changed (1) hide show
  1. video_processing.py +9 -4
video_processing.py CHANGED
@@ -203,11 +203,16 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
203
  if progress is not None:
204
  progress(0.95, desc="Generating video with heatmap")
205
 
206
- heatmap_video_path = create_video_with_heatmap(
207
- video_path, df, mse_embeddings, mse_posture, mse_voice,
208
- output_folder, desired_fps, largest_cluster
 
 
 
 
 
209
  )
210
-
211
  if heatmap_video_path is None:
212
  print("Failed to create heatmap video")
213
  else:
 
203
  if progress is not None:
204
  progress(0.95, desc="Generating video with heatmap")
205
 
206
+ heatmap_video_path = create_video_with_heatmap(
207
+ video_path,
208
+ df,
209
+ mse_embeddings,
210
+ mse_posture,
211
+ mse_voice,
212
+ output_folder,
213
+ analysis_fps
214
  )
215
+
216
  if heatmap_video_path is None:
217
  print("Failed to create heatmap video")
218
  else: