Update visualization.py

visualization.py

@@ -203,13 +203,6 @@ def plot_posture(df, posture_scores, color='blue', anomaly_threshold=3):
     plt.close()
     return fig
 
-def fig_to_img(fig):
-    canvas = FigureCanvasAgg(fig)
-    canvas.draw()
-    img = np.frombuffer(canvas.tostring_rgb(), dtype='uint8')
-    img = img.reshape(canvas.get_width_height()[::-1] + (3,))
-    plt.close(fig)
-    return img
 
 def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, output_path, desired_fps):
     cap = cv2.VideoCapture(video_path)