oskarastrom commited on
Commit
cfe7c49
·
1 Parent(s): ac53593

Update upload_ui.py

Browse files
Files changed (1) hide show
  1. gradio_scripts/upload_ui.py +6 -4
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
- settings.append(gr.Dropdown(label="Model", value=default_settings.find_model(models), choices=list(models.keys())))
 
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
- tracker = gr.Dropdown(["None", "Confidence Boost", "ByteTrack"], value=TrackerType.toString(default_settings.associative_tracker), label="Associative Tracking")
 
37
  settings.append(tracker)
38
- with gr.Row(visible=False) as track_row:
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=False) as track_row:
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)