Update visualization.py
Browse files- visualization.py +1 -1
visualization.py
CHANGED
@@ -201,7 +201,7 @@ def plot_posture(df, posture_scores, color='blue', anomaly_threshold=3):
|
|
201 |
return fig
|
202 |
|
203 |
|
204 |
-
def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture,
|
205 |
from matplotlib.backends.backend_agg import FigureCanvasAgg
|
206 |
# Open the video
|
207 |
cap = cv2.VideoCapture(video_path)
|
|
|
201 |
return fig
|
202 |
|
203 |
|
204 |
+
def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, output_path):
|
205 |
from matplotlib.backends.backend_agg import FigureCanvasAgg
|
206 |
# Open the video
|
207 |
cap = cv2.VideoCapture(video_path)
|