freealise commited on
Commit
7a82715
1 Parent(s): eee9496

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
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='pil')
298
- output_depth = gr.Image(label="Depth", type='pil')
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