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