rolpotamias commited on
Commit
12c1e62
·
verified ·
1 Parent(s): e20d68e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -2,8 +2,8 @@ import os
2
  import sys
3
  os.environ["PYOPENGL_PLATFORM"] = "egl"
4
  os.environ["MESA_GL_VERSION_OVERRIDE"] = "4.1"
5
- os.system('pip install /home/user/app/pyrender')
6
- sys.path.append('/home/user/app/pyrender')
7
 
8
  import gradio as gr
9
  import spaces
@@ -114,13 +114,15 @@ def run_wilow_model(image, conf, IoU_threshold=0.5):
114
  scene_bg_color=(1, 1, 1),
115
  focal_length=scaled_focal_length,
116
  )
117
- # cam_view = renderer.render_rgba_multiple(all_verts, cam_t=all_cam_t, render_res=img_size[n], is_right=all_right, **misc_args)
 
 
118
 
119
- # # Overlay image
120
 
121
- # input_img = img_vis.astype(np.float32)/255.0
122
- # input_img = np.concatenate([input_img, np.ones_like(input_img[:,:,:1])], axis=2) # Add alpha channel
123
- # input_img_overlay = input_img[:,:,:3] * (1-cam_view[:,:,3:]) + cam_view[:,:,:3] * cam_view[:,:,3:]
124
 
125
  image = img_vis #input_img_overlay
126
  return image, f'{len(detections)} hands detected'
 
2
  import sys
3
  os.environ["PYOPENGL_PLATFORM"] = "egl"
4
  os.environ["MESA_GL_VERSION_OVERRIDE"] = "4.1"
5
+ # os.system('pip install /home/user/app/pyrender')
6
+ # sys.path.append('/home/user/app/pyrender')
7
 
8
  import gradio as gr
9
  import spaces
 
114
  scene_bg_color=(1, 1, 1),
115
  focal_length=scaled_focal_length,
116
  )
117
+ print(all_verts[0])
118
+ print(all_cam_t[0])
119
+ cam_view = renderer.render_rgba_multiple(all_verts, cam_t=all_cam_t, render_res=img_size[n], is_right=all_right, **misc_args)
120
 
121
+ # Overlay image
122
 
123
+ input_img = img_vis.astype(np.float32)/255.0
124
+ input_img = np.concatenate([input_img, np.ones_like(input_img[:,:,:1])], axis=2) # Add alpha channel
125
+ input_img_overlay = input_img[:,:,:3] * (1-cam_view[:,:,3:]) + cam_view[:,:,:3] * cam_view[:,:,3:]
126
 
127
  image = img_vis #input_img_overlay
128
  return image, f'{len(detections)} hands detected'