diff --git a/mmdet3d/datasets/transforms/dbsampler.py b/mmdet3d/datasets/transforms/dbsampler.py index 56e8440b74..093cdfb170 100644 --- a/mmdet3d/datasets/transforms/dbsampler.py +++ b/mmdet3d/datasets/transforms/dbsampler.py @@ -280,7 +280,7 @@ def sample_all(self, s_points_list.append(s_points) gt_labels = np.array([self.cat2label[s['name']] for s in sampled], - dtype=np.long) + dtype=np.int64) if ground_plane is not None: xyz = sampled_gt_bboxes[:, :3] diff --git a/mmdet3d/visualization/local_visualizer.py b/mmdet3d/visualization/local_visualizer.py index d1eff4af97..1bc63701a5 100644 --- 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()