Spaces:
Runtime error
Runtime error
Commit ·
cfe7c49
1
Parent(s): ac53593
Update upload_ui.py
Browse files
gradio_scripts/upload_ui.py
CHANGED
|
@@ -21,7 +21,8 @@ def Upload_Gradio(gradio_components):
|
|
| 21 |
default_settings = InferenceConfig()
|
| 22 |
settings = []
|
| 23 |
with gr.Accordion("Advanced Settings", open=False):
|
| 24 |
-
|
|
|
|
| 25 |
|
| 26 |
gr.Markdown("Detection Parameters")
|
| 27 |
with gr.Row():
|
|
@@ -33,13 +34,14 @@ def Upload_Gradio(gradio_components):
|
|
| 33 |
settings.append(gr.Slider(0, 100, value=default_settings.min_hits, label="Min Hits", info="Minimum number of frames a fish has to appear in to count"))
|
| 34 |
settings.append(gr.Slider(0, 100, value=default_settings.max_age, label="Max Age", info="Max age of occlusion before track is split"))
|
| 35 |
|
| 36 |
-
|
|
|
|
| 37 |
settings.append(tracker)
|
| 38 |
-
with gr.Row(visible=
|
| 39 |
settings.append(gr.Slider(0, 5, value=default_settings.boost_power, label="Boost Power", info=""))
|
| 40 |
settings.append(gr.Slider(0, 1, value=default_settings.boost_decay, label="Boost Decay", info=""))
|
| 41 |
tracker.change(lambda x: gr.update(visible=(x=="Confidence Boost")), tracker, track_row)
|
| 42 |
-
with gr.Row(visible=
|
| 43 |
settings.append(gr.Slider(0, 1, value=default_settings.byte_low_conf, label="Low Conf Threshold", info=""))
|
| 44 |
settings.append(gr.Slider(0, 1, value=default_settings.byte_high_conf, label="High Conf Threshold", info=""))
|
| 45 |
tracker.change(lambda x: gr.update(visible=(x=="ByteTrack")), tracker, track_row)
|
|
|
|
| 21 |
default_settings = InferenceConfig()
|
| 22 |
settings = []
|
| 23 |
with gr.Accordion("Advanced Settings", open=False):
|
| 24 |
+
default_model = default_settings.find_model(models)
|
| 25 |
+
settings.append(gr.Dropdown(label="Model", value=default_model, choices=list(models.keys())))
|
| 26 |
|
| 27 |
gr.Markdown("Detection Parameters")
|
| 28 |
with gr.Row():
|
|
|
|
| 34 |
settings.append(gr.Slider(0, 100, value=default_settings.min_hits, label="Min Hits", info="Minimum number of frames a fish has to appear in to count"))
|
| 35 |
settings.append(gr.Slider(0, 100, value=default_settings.max_age, label="Max Age", info="Max age of occlusion before track is split"))
|
| 36 |
|
| 37 |
+
default_tracker = TrackerType.toString(default_settings.associative_tracker)
|
| 38 |
+
tracker = gr.Dropdown(["None", "Confidence Boost", "ByteTrack"], value=default_tracker, label="Associative Tracking")
|
| 39 |
settings.append(tracker)
|
| 40 |
+
with gr.Row(visible=default_tracker=="Confidence Boost") as track_row:
|
| 41 |
settings.append(gr.Slider(0, 5, value=default_settings.boost_power, label="Boost Power", info=""))
|
| 42 |
settings.append(gr.Slider(0, 1, value=default_settings.boost_decay, label="Boost Decay", info=""))
|
| 43 |
tracker.change(lambda x: gr.update(visible=(x=="Confidence Boost")), tracker, track_row)
|
| 44 |
+
with gr.Row(visible=default_tracker=="ByteTrack") as track_row:
|
| 45 |
settings.append(gr.Slider(0, 1, value=default_settings.byte_low_conf, label="Low Conf Threshold", info=""))
|
| 46 |
settings.append(gr.Slider(0, 1, value=default_settings.byte_high_conf, label="High Conf Threshold", info=""))
|
| 47 |
tracker.change(lambda x: gr.update(visible=(x=="ByteTrack")), tracker, track_row)
|