Spaces:
Runtime error
Runtime error
get rid of the color, so the maps are usable
#5
by
Hugginguser123456789000
- opened
app.py
CHANGED
|
@@ -96,7 +96,7 @@ def depth_normal(img, model_selection="vit-small"):
|
|
| 96 |
pred_color_normal = vis_surface_normal(pred_normal)
|
| 97 |
|
| 98 |
##formatted = (output * 255 / np.max(output)).astype('uint8')
|
| 99 |
-
img = Image.fromarray(
|
| 100 |
img_normal = Image.fromarray(pred_color_normal)
|
| 101 |
return img, img_normal
|
| 102 |
|
|
|
|
| 96 |
pred_color_normal = vis_surface_normal(pred_normal)
|
| 97 |
|
| 98 |
##formatted = (output * 255 / np.max(output)).astype('uint8')
|
| 99 |
+
img = Image.fromarray(pred_normal)
|
| 100 |
img_normal = Image.fromarray(pred_color_normal)
|
| 101 |
return img, img_normal
|
| 102 |
|