Spaces:

NN-BRD
/

hackathon_depth_segment

Runtime error

jens commited on Aug 16, 2023

Commit

de13e02

1 Parent(s): 9dd1448

fix

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -64,7 +64,7 @@ class DepthPredictor:
         formatted = 255 - (output * 255 / np.max(output)).astype('uint8')
         #img = Image.fromarray(formatted)
-        return 1 - output
     def generate_pcl(self, image):
         print(np.array(image).shape)

         formatted = 255 - (output * 255 / np.max(output)).astype('uint8')
         #img = Image.fromarray(formatted)
+        return output
     def generate_pcl(self, image):
         print(np.array(image).shape)