jens commited on
Commit
2ab737a
·
1 Parent(s): d03ff7c
Files changed (1) hide show
  1. inference.py +3 -4
inference.py CHANGED
@@ -81,10 +81,9 @@ class DepthPredictor:
81
  mode="bicubic",
82
  align_corners=False,
83
  ).squeeze()
84
-
85
- output = 1 - (prediction.cpu().numpy() / np.max(prediction))
86
- #formatted = 255 - (output * 255 / np.max(output)).astype('uint8')
87
- #img = Image.fromarray(formatted)
88
  return output
89
 
90
  def generate_pcl(self, image):
 
81
  mode="bicubic",
82
  align_corners=False,
83
  ).squeeze()
84
+
85
+ output = prediction.cpu().numpy()
86
+ output = 1 - (output/np.max(output))
 
87
  return output
88
 
89
  def generate_pcl(self, image):