Update video_processing.py

video_processing.py

@@ -23,7 +23,6 @@ mtcnn = MTCNN(keep_all=False, device=device, thresholds=[0.9, 0.9, 0.9], min_fac
 mp_face_mesh = mp.solutions.face_mesh
 face_mesh = mp_face_mesh.FaceMesh(static_image_mode=False, max_num_faces=1, min_detection_confidence=0.8)
 
-
 def extract_frames(video_path, output_folder, desired_fps, progress_callback=None):
     os.makedirs(output_folder, exist_ok=True)
     clip = VideoFileClip(video_path)
@@ -199,7 +198,6 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             mse_heatmap_voice = plot_mse_heatmap(mse_voice, "Voice MSE Heatmap", df)
 
             correlation_heatmap = plot_correlation_heatmap(mse_embeddings, mse_posture, mse_voice)
-            scatter_plot_3d = plot_3d_scatter(mse_embeddings, mse_posture, mse_voice)
 
             try:
                 if progress is not None:
@@ -228,7 +226,7 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             print(f"Error details: {str(e)}")
             import traceback
             traceback.print_exc()
-            return (f"Error in video processing: {str(e)}",) + (None,) * 25
+            return (f"Error in video processing: {str(e)}",) + (None,) * 24
 
         progress(1.0, "Preparing results")
         results = f"Number of persons detected: {num_clusters}\n\n"
@@ -296,8 +294,7 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             aligned_faces_folder,
             frames_folder,
             heatmap_video_path,
-            correlation_heatmap,
-            scatter_plot_3d
+            correlation_heatmap
         )