Spaces:
Runtime error
Runtime error
jens
commited on
Commit
·
2ab737a
1
Parent(s):
d03ff7c
fix
Browse files- 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 =
|
86 |
-
|
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):
|