Spaces:
Running
on
T4
Running
on
T4
fix normal.npy file save mistake
Browse files
app.py
CHANGED
@@ -116,7 +116,7 @@ def predict_depth_normal(img, model_selection="vit-small", fx=1000.0, fy=1000.0,
|
|
116 |
depth_file = f"{state_cache['save_dir']}/depth.npy"
|
117 |
normal_file = f"{state_cache['save_dir']}/normal.npy"
|
118 |
np.save(depth_file, pred_depth)
|
119 |
-
np.save(normal_file,
|
120 |
|
121 |
##formatted = (output * 255 / np.max(output)).astype('uint8')
|
122 |
img = Image.fromarray(pred_color)
|
|
|
116 |
depth_file = f"{state_cache['save_dir']}/depth.npy"
|
117 |
normal_file = f"{state_cache['save_dir']}/normal.npy"
|
118 |
np.save(depth_file, pred_depth)
|
119 |
+
np.save(normal_file, pred_normal)
|
120 |
|
121 |
##formatted = (output * 255 / np.max(output)).astype('uint8')
|
122 |
img = Image.fromarray(pred_color)
|