Hancy commited on
Commit
2014fad
Β·
1 Parent(s): d10ef69
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -54,25 +54,25 @@ with gr.Blocks(css=CSS) as demo:
54
  output_image = gr.Image(label="Output Range Map", elem_id='img-display-output')
55
  output_pcd = gr.Model3D(label="Output Point Cloud", elem_id='pcd-display-output', interactive=False)
56
 
57
- raw_file = gr.File(label="Point Cloud (.txt file). Can be viewed through Meshlab")
58
  submit = gr.Button("Generate")
59
 
60
  def on_submit(image):
61
  cond = load_camera(image)
62
  img, pcd = generate_lidar(model, cond)
63
 
64
- tmp = tempfile.NamedTemporaryFile(suffix='.txt', delete=False)
65
- pcd.save(tmp.name)
66
 
67
  rgb_img = colorize_depth(img, log_scale=True)
68
 
69
- return [rgb_img, pcd, tmp.name]
70
 
71
- submit.click(on_submit, inputs=[input_image], outputs=[output_image, output_pcd, raw_file])
72
 
73
  example_files = sorted(os.listdir('cam_examples'))
74
  example_files = [os.path.join('cam_examples', filename) for filename in example_files]
75
- examples = gr.Examples(examples=example_files, inputs=[input_image], outputs=[output_image, output_pcd, raw_file],
76
  fn=on_submit, cache_examples=False)
77
 
78
  gr.Markdown(FOOTER)
 
54
  output_image = gr.Image(label="Output Range Map", elem_id='img-display-output')
55
  output_pcd = gr.Model3D(label="Output Point Cloud", elem_id='pcd-display-output', interactive=False)
56
 
57
+ # raw_file = gr.File(label="Point Cloud (.txt file). Can be viewed through Meshlab")
58
  submit = gr.Button("Generate")
59
 
60
  def on_submit(image):
61
  cond = load_camera(image)
62
  img, pcd = generate_lidar(model, cond)
63
 
64
+ # tmp = tempfile.NamedTemporaryFile(suffix='.txt', delete=False)
65
+ # pcd.save(tmp.name)
66
 
67
  rgb_img = colorize_depth(img, log_scale=True)
68
 
69
+ return [rgb_img, pcd]
70
 
71
+ submit.click(on_submit, inputs=[input_image], outputs=[output_image, output_pcd])
72
 
73
  example_files = sorted(os.listdir('cam_examples'))
74
  example_files = [os.path.join('cam_examples', filename) for filename in example_files]
75
+ examples = gr.Examples(examples=example_files, inputs=[input_image], outputs=[output_image, output_pcd],
76
  fn=on_submit, cache_examples=False)
77
 
78
  gr.Markdown(FOOTER)