Hugginguser123456789000 commited on
Commit
d727589
1 Parent(s): cdd280b

get rid of the color, so the maps are usable

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