Spaces:
Running
on
A10G
Running
on
A10G
Commit
·
2280b70
1
Parent(s):
0531772
Update requirements.txt
Browse files- app.py +4 -4
- 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
|
| 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=
|
| 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=[
|
| 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
|