reab5555 commited on
Commit
ecf60ef
·
verified ·
1 Parent(s): 27f2e76

Update video_processing.py

Browse files
Files changed (1) hide show
  1. video_processing.py +5 -4
video_processing.py CHANGED
@@ -202,21 +202,22 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
202
  try:
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
  if heatmap_video_path is None:
216
  print("Failed to create heatmap video")
217
  else:
218
  print(f"Heatmap video path from create_video_with_heatmap: {heatmap_video_path}")
219
-
220
  if progress is not None:
221
  progress(1.0, "Video processing complete")
222
 
 
202
  try:
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:
219
  print(f"Heatmap video path from create_video_with_heatmap: {heatmap_video_path}")
220
+
221
  if progress is not None:
222
  progress(1.0, "Video processing complete")
223