Update video_processing.py
Browse files- video_processing.py +1 -9
video_processing.py
CHANGED
|
@@ -203,15 +203,7 @@ 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 |
-
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 |
-
original_fps
|
| 214 |
-
)
|
| 215 |
|
| 216 |
if heatmap_video_path is None:
|
| 217 |
print("Failed to create heatmap video")
|
|
|
|
| 203 |
if progress is not None:
|
| 204 |
progress(0.95, desc="Generating video with heatmap")
|
| 205 |
|
| 206 |
+
heatmap_video_path = create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_voice, output_folder, desired_fps, largest_cluster)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 207 |
|
| 208 |
if heatmap_video_path is None:
|
| 209 |
print("Failed to create heatmap video")
|