Skip to content

Update demo_gradio.py #60

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
magickenshin wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Update demo_gradio.py #60

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
57 changes: 57 additions & 0 deletions demo_gradio.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -313,6 +313,15 @@ def update_visualization(
return glbfile, "Updating Visualization"


def check_target_dir(target_dir):
"""
Check if the target_dir exists.
"""
if not os.path.isdir(target_dir):
return "None"
return target_dir


# -------------------------------------------------------------------------
# Example images
# -------------------------------------------------------------------------
Expand Down Expand Up @@ -449,6 +458,7 @@ def update_visualization(
[input_video, input_images, reconstruction_output, log_output, target_dir_output, image_gallery],
scale=1,
)
reload_btn = gr.Button("Reload", scale=1, variant="secondary")

with gr.Row():
prediction_mode = gr.Radio(
Expand All @@ -459,6 +469,14 @@ def update_visualization(
elem_id="my_radio",
)

with gr.Row():
gr.Interface(
fn=check_target_dir,
inputs=[gr.Textbox(label="Input Existing target_dir", value="/path/to/target_dir")],
outputs=[target_dir_output],
flagging_mode="never",
)

with gr.Row():
conf_thres = gr.Slider(minimum=0, maximum=100, value=50, step=0.1, label="Confidence Threshold (%)")
frame_filter = gr.Dropdown(choices=["All"], value="All", label="Show Points from Frame")
Expand Down Expand Up @@ -559,6 +577,45 @@ def example_pipeline(
fn=lambda: "False", inputs=[], outputs=[is_example] # set is_example to "False"
)

# -------------------------------------------------------------------------
# "Reload" button logic:
# - Clear fields
# - Check prediction
# - Then update visualization
# -------------------------------------------------------------------------
def check_prediction(target_dir):
if not os.path.exists(target_dir):
is_example = "True"
return "None", is_example
else:
is_example = "False"
return target_dir, is_example

reload_btn.click(fn=clear_fields, inputs=[], outputs=[]).then(
fn=update_log, inputs=[], outputs=[log_output]
).then(
fn=check_prediction,
inputs=[target_dir_output],
outputs=[
target_dir_output,
is_example,
]
).then(
update_visualization,
inputs=[
target_dir_output,
conf_thres,
frame_filter,
mask_black_bg,
mask_white_bg,
show_cam,
mask_sky,
prediction_mode,
is_example,
],
outputs=[reconstruction_output, log_output],
)

# -------------------------------------------------------------------------
# Real-time Visualization Updates
# -------------------------------------------------------------------------
Expand Down