Spaces:
Running
on
A10G
Running
on
A10G
pookiefoof
commited on
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
|