remove prints

2023-12-13 22:18:58 +02:00
parent 30049bc659
commit d5529f69d5
1 changed files with 0 additions and 2 deletions
@@ -170,9 +170,7 @@ class ColorizeDepthmap:
    def color(self, image, colorize_method):
        colored_images = []
        for i in range(image.shape[0]):  # Iterate over the batch dimension
-            print(image[i].shape)
            depth_map = image[i].squeeze().permute(2, 0, 1)
-            print(depth_map.shape)
            depth_map = depth_map[0]
            depth_map = colorizedepth(depth_map, colorize_method)
            depth_map = torch.from_numpy(depth_map) / 255