Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
|
|
|
|
118 |
|
119 |
-
|
120 |
|
121 |
-
|
122 |
-
|
123 |
-
|
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'
|