Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -209,10 +209,6 @@ def pano_depth_to_world_points(depth):
|
|
209 |
return pts3d
|
210 |
|
211 |
def get_mesh(image, depth, keep_edges=False):
|
212 |
-
image.thumbnail((518,518)) # limit the size of the image
|
213 |
-
depth.thumbnail((518,518))
|
214 |
-
image = np.array(image)
|
215 |
-
depth = np.array(depth)
|
216 |
pts3d = pano_depth_to_world_points(depth)
|
217 |
print('radius from depth - ok')
|
218 |
|
@@ -294,8 +290,8 @@ with gr.Blocks(css=css) as demo:
|
|
294 |
checkbox = gr.Checkbox(label="Keep occlusion edges", value=True)
|
295 |
processed_video = gr.Video(label="Output Video", format="mp4")
|
296 |
processed_zip = gr.File(label="Output Archive")
|
297 |
-
output_frame = gr.Image(label="Frame", type='
|
298 |
-
output_depth = gr.Image(label="Depth", type='
|
299 |
result = gr.Model3D(label="3D Mesh", clear_color=[
|
300 |
1.0, 1.0, 1.0, 1.0])
|
301 |
|
|
|
209 |
return pts3d
|
210 |
|
211 |
def get_mesh(image, depth, keep_edges=False):
|
|
|
|
|
|
|
|
|
212 |
pts3d = pano_depth_to_world_points(depth)
|
213 |
print('radius from depth - ok')
|
214 |
|
|
|
290 |
checkbox = gr.Checkbox(label="Keep occlusion edges", value=True)
|
291 |
processed_video = gr.Video(label="Output Video", format="mp4")
|
292 |
processed_zip = gr.File(label="Output Archive")
|
293 |
+
output_frame = gr.Image(label="Frame", type='numpy')
|
294 |
+
output_depth = gr.Image(label="Depth", type='numpy')
|
295 |
result = gr.Model3D(label="3D Mesh", clear_color=[
|
296 |
1.0, 1.0, 1.0, 1.0])
|
297 |
|