Spaces:

NN-BRD
/

hackathon_depth_segment

Runtime error

jens commited on Aug 17, 2023

Commit

d03ff7c

1 Parent(s): 32a0a1c

fix

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -82,7 +82,7 @@ class DepthPredictor:
                                 align_corners=False,
                         ).squeeze()
-        output = 1 - (prediction.cpu().numpy() / np.max(output))
         #formatted = 255 - (output * 255 / np.max(output)).astype('uint8')
         #img = Image.fromarray(formatted)
         return output

                                 align_corners=False,
                         ).squeeze()
+        output = 1 - (prediction.cpu().numpy() / np.max(prediction))
         #formatted = 255 - (output * 255 / np.max(output)).astype('uint8')
         #img = Image.fromarray(formatted)
         return output