Spaces:
Running
on
Zero
Running
on
Zero
File size: 4,794 Bytes
184193d 6b1bc1c 184193d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 |
import os
import glob
import gradio as gr
from functools import partial
from PIL import Image
from .gradio_custommodel3d import CustomModel3D
from .gradio_customgs import CustomGS
def create_interface_views_to_3d(freesplatter_api):
example_root = 'examples/views_to_3d'
examples = []
for dir in os.listdir(example_root):
sample_dir = os.path.join(example_root, dir)
input_files = sorted(glob.glob(f'{sample_dir}/*.png')+glob.glob(f'{sample_dir}/*.jpg'))
examples.append(input_files)
var_dict = dict()
with gr.Blocks(analytics_enabled=False) as interface:
with gr.Row():
with gr.Column(scale=1):
with gr.Row():
var_dict['image_files'] = gr.Gallery(
label='Input images',
type="filepath",
file_types=['image'],
show_label=False,
columns=[5],
rows=[2],
object_fit='contain',
height="auto",
)
with gr.Accordion("Reconstruction settings", open=True):
with gr.Row():
var_dict['do_rembg'] = gr.Checkbox(
label='Remove background',
value=False,
container=False,
)
with gr.Row():
var_dict['gs_type'] = gr.Radio(
choices=['2DGS', '3DGS'],
value='2DGS',
type='value',
label='Gaussian splatting type',
info='2DGS often leads to better mesh geometry',
)
var_dict['mesh_reduction'] = gr.Slider(
label="Mesh simplification ratio",
info='Larger ratio leads to less faces',
minimum=0.7,
maximum=0.95,
value=0.8,
step=0.05,
)
with gr.Row(equal_height=False):
var_dict['run_btn'] = gr.Button('Reconstruct', variant='primary', scale=2)
with gr.Row():
var_dict['out_multiview'] = gr.Image(
label='Input views',
interactive=False,
visible=False,
)
snapshot_1 = gr.Image(None, visible=False, image_mode='RGBA')
snapshot_2 = gr.Image(None, visible=False, image_mode='RGBA')
snapshot_3 = gr.Image(None, visible=False, image_mode='RGBA')
snapshot_4 = gr.Image(None, visible=False, image_mode='RGBA')
def set_gallery_images(*images):
return list(images)
gr.Examples(
examples=examples,
fn=set_gallery_images,
inputs=[snapshot_1, snapshot_2, snapshot_3, snapshot_4],
outputs=[var_dict['image_files']],
run_on_click=True,
cache_examples=False,
label='Examples (click one of the rows below to start)',
examples_per_page=5,
)
with gr.Column(scale=1):
var_dict['out_pose'] = gr.Plot(
label='Estimated poses',
)
var_dict['out_gs_vis'] = CustomGS(
label='Output GS',
interactive=False,
height=320,
)
var_dict['out_video'] = gr.Video(
label='Output video',
interactive=False,
autoplay=True,
height=320,
)
var_dict['out_mesh'] = CustomModel3D(
label='Output mesh',
interactive=False,
height=400,
)
var_dict['run_btn'].click(
fn=partial(freesplatter_api, cache_dir=interface.GRADIO_CACHE),
inputs=[var_dict['image_files'],
var_dict['do_rembg'],
var_dict['gs_type'],
var_dict['mesh_reduction']],
outputs=[var_dict['out_multiview'], var_dict['out_gs_vis'], var_dict['out_video'], var_dict['out_mesh'], var_dict['out_pose']],
concurrency_id='default_group',
api_name='run_views_to_3d',
)
return interface, var_dict
|