From d5529f69d51bb4f0d7d49a159fabefb36b659a56 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Wed, 13 Dec 2023 22:18:58 +0200 Subject: [PATCH] remove prints --- nodes.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/nodes.py b/nodes.py index d35a2be..5f5a3f2 100644 --- a/nodes.py +++ b/nodes.py @@ -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