Fix the problem of failing to save images after open3d visualization.

open-mmlab · Nov 21, 2024 · 934bcb5 · 934bcb5
1 parent 962f093
commit 934bcb5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmdet3d/visualization/local_visualizer.py b/mmdet3d/visualization/local_visualizer.py
@@ -884,7 +884,7 @@ def show(self,
                     self.view_port = \
                         self.view_control.convert_to_pinhole_camera_parameters()  # noqa: E501
                 self.flag_next = False
-            self.o3d_vis.clear_geometries()
+            # self.o3d_vis.clear_geometries()
             try:
                 del self.pcd
             except (KeyError, AttributeError):
@@ -893,7 +893,7 @@ def show(self,
                 if not (save_path.endswith('.png')
                         or save_path.endswith('.jpg')):
                     save_path += '.png'
-                self.o3d_vis.capture_screen_image(save_path)
+                self.o3d_vis.capture_screen_image(save_path, do_render=True)
             if self.flag_exit:
                 self.o3d_vis.destroy_window()
                 self.o3d_vis.close()