Close figures once saved

rameau-fr · Jun 12, 2024 · 5e28b01 · 5e28b01
1 parent 773ef21
commit 5e28b01
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 5 deletions.
diff --git a/python_utils/compute_error_statistic.py b/python_utils/compute_error_statistic.py
@@ -34,8 +34,8 @@ def visualize_and_save_results(
     plt.ylabel("Mean reprojection error")
     plt.title("Mean error per frame")
     plt.show()
-
     plt.savefig(reprojection_error_data_path / "mean_reprojection_error_per_frame.png")
+    plt.close()
 
 
 def compute_error_statistic(reprojection_error_data_path: Path) -> None:

diff --git a/python_utils/compute_pose_error_vs_gt.py b/python_utils/compute_pose_error_vs_gt.py
@@ -191,8 +191,9 @@ def compute_pose_error_vs_gt(
     ax.set_xlabel("X")
     ax.set_ylabel("Y")
     ax.set_zlabel("Z")
-    plt.savefig(calibrated_cameras_data.parent / "pose_error_vs_gt.png")
     plt.show()
+    plt.savefig(calibrated_cameras_data.parent / "pose_error_vs_gt.png")
+    plt.close()
 
 
 if __name__ == "__main__":

diff --git a/python_utils/display_calib_cameras.py b/python_utils/display_calib_cameras.py
@@ -265,8 +265,8 @@ def display_calib_cameras(calib_cameras_data_path: Path) -> None:
 
     prepare_figure(ax)
     plt.show()
-
     plt.savefig(calib_cameras_data_path.parent / "cameras_calib.png")
+    plt.close()
 
 
 if __name__ == "__main__":

diff --git a/python_utils/display_calib_object.py b/python_utils/display_calib_object.py
@@ -26,8 +26,8 @@ def display_calib_object(calib_object_data_path: Path) -> None:
 
     prepare_figure(ax)
     plt.show()
-
     plt.savefig(calib_object_data_path.parent / "calibrated_object.png")
+    plt.close()
 
 
 if __name__ == "__main__":

diff --git a/python_utils/display_cam_obj_pose.py b/python_utils/display_cam_obj_pose.py
@@ -54,8 +54,8 @@ def display_cam_obj_pose(calib_data: Path) -> None:
 
     axis_equal_3d(ax)
     plt.show()
-
     plt.savefig(calib_data / "cam_obj_pose.png")
+    plt.close()
 
 
 if __name__ == "__main__":