JUGGHM commited on
Commit
a5b7ecf
1 Parent(s): b1a4c74

fix normal.npy file save mistake

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, pred_depth)
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)