pookiefoof commited on
Commit
2280b70
1 Parent(s): 0531772

Update requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +4 -4
  2. requirements.txt +1 -0
app.py CHANGED
@@ -370,7 +370,7 @@ class Inference3D_API:
370
  material.pbrMetallicRoughness.roughnessFactor = 1.0
371
  gltf.save(f'{save_dir}/output.glb')
372
 
373
- return save_dir, f"{save_dir}/output.obj", f"{save_dir}/output.glb"
374
 
375
  @torch.no_grad()
376
  def main(
@@ -420,11 +420,11 @@ def main(
420
  img_input2 = gr.Image(type="pil", label="Right Image", image_mode="RGBA", width=256, height=384)
421
  img_input3 = gr.Image(type="pil", label="Left Image", image_mode="RGBA", width=256, height=384)
422
  with gr.Column(variant="panel"):
423
- smooth_iter = gr.Slider(minimum=0, maximum=10, step=1, value=5, label="Laplacian Smoothing Iterations")
424
  with gr.Row():
425
  back_proj = gr.Checkbox(label="Back Projection")
426
  button2 = gr.Button(value="Generate 3D Mesh")
427
- output_dir = gr.Textbox(label="Output Directory")
428
  with gr.Row():
429
  with gr.Tab("GLB"):
430
  output_model_glb = gr.Model3D( label="Output Model (GLB Format)", height=512)
@@ -440,7 +440,7 @@ def main(
440
  button2.click(
441
  infer3dapi.process_images,
442
  inputs=[img_input0, img_input1, img_input2, img_input3, back_proj, smooth_iter],
443
- outputs=[output_dir, output_model_obj, output_model_glb]
444
  )
445
  demo.launch()
446
 
 
370
  material.pbrMetallicRoughness.roughnessFactor = 1.0
371
  gltf.save(f'{save_dir}/output.glb')
372
 
373
+ return f"{save_dir}/output.obj", f"{save_dir}/output.glb"
374
 
375
  @torch.no_grad()
376
  def main(
 
420
  img_input2 = gr.Image(type="pil", label="Right Image", image_mode="RGBA", width=256, height=384)
421
  img_input3 = gr.Image(type="pil", label="Left Image", image_mode="RGBA", width=256, height=384)
422
  with gr.Column(variant="panel"):
423
+ smooth_iter = gr.Slider(minimum=0, maximum=5, step=1, value=3, label="Laplacian Smoothing Iterations")
424
  with gr.Row():
425
  back_proj = gr.Checkbox(label="Back Projection")
426
  button2 = gr.Button(value="Generate 3D Mesh")
427
+ # output_dir = gr.Textbox(label="Output Directory")
428
  with gr.Row():
429
  with gr.Tab("GLB"):
430
  output_model_glb = gr.Model3D( label="Output Model (GLB Format)", height=512)
 
440
  button2.click(
441
  infer3dapi.process_images,
442
  inputs=[img_input0, img_input1, img_input2, img_input3, back_proj, smooth_iter],
443
+ outputs=[output_model_obj, output_model_glb]
444
  )
445
  demo.launch()
446
 
requirements.txt CHANGED
@@ -22,3 +22,4 @@ open3d
22
  trimesh
23
  pymeshlab
24
  pygltflib
 
 
22
  trimesh
23
  pymeshlab
24
  pygltflib
25
+ omegaconf