Update video_processing.py
Browse files- 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 |
-
|
207 |
-
video_path,
|
208 |
-
|
|
|
|
|
|
|
|
|
|
|
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:
|