reab5555 commited on
Commit
7090b9b
·
verified ·
1 Parent(s): c9bbbb9

Update video_processing.py

Browse files
Files changed (1) hide show
  1. video_processing.py +3 -2
video_processing.py CHANGED
@@ -183,8 +183,9 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
183
  progress(0.9, "Generating video with heatmap")
184
 
185
  # Create video with heatmap
186
- heatmap_video_path = os.path.join(output_folder, "heatmap_video.mp4")
187
- heatmap_video_path = create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, heatmap_video_path, desired_fps, largest_cluster)
 
188
 
189
  except Exception as e:
190
  print(f"Error details: {str(e)}")
 
183
  progress(0.9, "Generating video with heatmap")
184
 
185
  # Create video with heatmap
186
+ heatmap_video_path = os.path.join(output_folder, "video_with_heatmap.mp4")
187
+ create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, heatmap_video_path, desired_fps)
188
+
189
 
190
  except Exception as e:
191
  print(f"Error details: {str(e)}")