Update video_processing.py
Browse files- 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 |
-
|
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 |
|