Spaces:
Sleeping
Sleeping
Hugginguser123456789000
commited on
get rid of the color, so the maps are usable
Browse files
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 |
|