Sandeepa's picture
check for build fix
3078f9c
raw
history blame
501 Bytes
import gradio as gr
from depth_detection import depth_detection
from mesh_network import mesh_network
from point_cloud import point_cloud
def reconstruction3d(input_image):
new_image, output = depth_detection(input_image,pad=16)
pcd_img = point_cloud(new_image, output)
output_mesh = mesh_network(pcd_img)
return output_mesh
input_image = gr.inputs.Image(label="Input Image", type="pil")
iface = gr.Interface(fn=reconstruction3d, inputs=input_image, outputs="image")
iface.launch()