depth adjusted

inference.py

@@ -36,10 +36,10 @@ class DepthPredictor:
         output = prediction.cpu().numpy()
         formatted = (output * 255 / np.max(output)).astype('uint8')
         #img = Image.fromarray(formatted)
-        return formatted
+        return output
     
     def generate_pcl(self, image):
-        depth = 255 - self.predict(image)
+        depth = self.predict(image)
         # Step 2: Create an RGBD image from the RGB and depth image
         depth_o3d = o3d.geometry.Image(depth)
         image_o3d = o3d.geometry.Image(np.array(image))