Update video_processing.py
Browse files- video_processing.py +2 -1
video_processing.py
CHANGED
@@ -214,9 +214,10 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
|
|
214 |
if progress is not None:
|
215 |
progress(0.95, desc="Generating video with heatmap")
|
216 |
|
|
|
217 |
heatmap_video_path = create_video_with_heatmap(
|
218 |
video_path, df, mse_embeddings, mse_posture, mse_voice,
|
219 |
-
|
220 |
progress=progress
|
221 |
)
|
222 |
|
|
|
214 |
if progress is not None:
|
215 |
progress(0.95, desc="Generating video with heatmap")
|
216 |
|
217 |
+
output_folder = "output" # You can change this to your desired output folder
|
218 |
heatmap_video_path = create_video_with_heatmap(
|
219 |
video_path, df, mse_embeddings, mse_posture, mse_voice,
|
220 |
+
output_folder, original_fps, largest_cluster,
|
221 |
progress=progress
|
222 |
)
|
223 |
|