Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -72,7 +72,7 @@ def depth_normal(img):
|
|
72 |
pred_color = gray_to_colormap(pred_depth)
|
73 |
|
74 |
pred_normal = pred_normal.squeeze()
|
75 |
-
if pred_normal.size(0) == 3
|
76 |
pred_normal == pred_normal.permute(1,2,0)
|
77 |
pred_color_normal = vis_surface_normal(pred_normal)
|
78 |
|
|
|
72 |
pred_color = gray_to_colormap(pred_depth)
|
73 |
|
74 |
pred_normal = pred_normal.squeeze()
|
75 |
+
if pred_normal.size(0) == 3:
|
76 |
pred_normal == pred_normal.permute(1,2,0)
|
77 |
pred_color_normal = vis_surface_normal(pred_normal)
|
78 |
|