Spaces:
Runtime error
Runtime error
jens
commited on
Commit
·
5f53c0c
1
Parent(s):
17dfbdb
invert depth
Browse files- inference.py +2 -1
inference.py
CHANGED
|
@@ -34,7 +34,8 @@ class DepthPredictor:
|
|
| 34 |
).squeeze()
|
| 35 |
|
| 36 |
output = prediction.cpu().numpy()
|
| 37 |
-
formatted = (output * 255 / np.max(output)).astype('uint8')
|
|
|
|
| 38 |
#img = Image.fromarray(formatted)
|
| 39 |
return formatted
|
| 40 |
|
|
|
|
| 34 |
).squeeze()
|
| 35 |
|
| 36 |
output = prediction.cpu().numpy()
|
| 37 |
+
formatted = 255 - (output * 255 / np.max(output)).astype('uint8')
|
| 38 |
+
|
| 39 |
#img = Image.fromarray(formatted)
|
| 40 |
return formatted
|
| 41 |
|