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):
|