Autoload parameters

InferenceConfig.py

@@ -5,6 +5,11 @@ class TrackerType(Enum):
     CONF_BOOST = 1
     BYTETRACK = 2
 
+    def toString(val):
+        if val == TrackerType.NONE: return "None"
+        if val == TrackerType.CONF_BOOST: return "Confidence Boost"
+        if val == TrackerType.BYTETRACK: return "ByteTrack"
+
 ### Configuration options
 WEIGHTS = 'models/v5m_896_300best.pt'
 # will need to configure these based on GPU hardware
@@ -16,7 +21,7 @@ MAX_AGE = 20 # time until missing fish get's new id
 MIN_HITS = 11 # minimum number of frames with a specific fish for it to count
 MIN_LENGTH = 0.3 # minimum fish length, in meters
 IOU_THRES = 0.01 # IOU threshold for tracking
-MIN_TRAVEL = -1 # Minimum distance a track has to travel
+MIN_TRAVEL = 0 # Minimum distance a track has to travel
 DEFAULT_TRACKER = TrackerType.BYTETRACK
 
 class InferenceConfig:
@@ -50,6 +55,13 @@ class InferenceConfig:
         self.byte_low_conf = low
         self.byte_high_conf = high
 
+    def find_model(self, model_list):
+        for model_name, model_path in enumerate(model_list):
+            if model_path == self.weights:
+                return model_name
+        return None
+
+
     def to_dict(self):
         dict = {
             'weights': self.weights,

gradio_scripts/upload_ui.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from gradio_scripts.file_reader import File
+from InferenceConfig import InferenceConfig, TrackerType
 
 
 models = {
@@ -17,35 +18,36 @@ def Upload_Gradio(gradio_components):
 
             gr.HTML("<p align='center' style='font-size: large;font-style: italic;'>Submit an .aris file to analyze result.</p>")
             
+            default_settings = InferenceConfig()
             settings = []
             with gr.Accordion("Advanced Settings", open=False):
-                settings.append(gr.Dropdown(label="Model", value="master", choices=list(models.keys())))
+                settings.append(gr.Dropdown(label="Model", value=default_settings.find_model(models), choices=list(models.keys())))
 
                 gr.Markdown("Detection Parameters")
                 with gr.Row():
-                    settings.append(gr.Slider(0, 1, value=0.05, label="Confidence Threshold", info="Confidence cutoff for detection boxes"))
-                    settings.append(gr.Slider(0, 1, value=0.2, label="NMS IoU", info="IoU threshold for non-max suppression"))
+                    settings.append(gr.Slider(0, 1, value=default_settings.conf_thresh, label="Confidence Threshold", info="Confidence cutoff for detection boxes"))
+                    settings.append(gr.Slider(0, 1, value=default_settings.nms_iou, label="NMS IoU", info="IoU threshold for non-max suppression"))
 
                 gr.Markdown("Tracking Parameters")
                 with gr.Row():
-                    settings.append(gr.Slider(0, 100, value=16, label="Min Hits", info="Minimum number of frames a fish has to appear in to count"))
-                    settings.append(gr.Slider(0, 100, value=14, label="Max Age", info="Max age of occlusion before track is split"))
+                    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"))
+                    settings.append(gr.Slider(0, 100, value=default_settings.max_age, label="Max Age", info="Max age of occlusion before track is split"))
 
-                tracker = gr.Dropdown(["None", "Confidence Boost", "ByteTrack"], label="Associative Tracking", value="None")
+                tracker = gr.Dropdown(["None", "Confidence Boost", "ByteTrack"], value=TrackerType.toString(default_settings.associative_tracker), label="Associative Tracking")
                 settings.append(tracker)
                 with gr.Row(visible=False) as track_row:
-                    settings.append(gr.Slider(0, 5, value=1, label="Boost Power", info=""))
-                    settings.append(gr.Slider(0, 1, value=1, label="Boost Decay", info=""))
+                    settings.append(gr.Slider(0, 5, value=default_settings.boost_power, label="Boost Power", info=""))
+                    settings.append(gr.Slider(0, 1, value=default_settings.boost_decay, label="Boost Decay", info=""))
                     tracker.change(lambda x:  gr.update(visible=(x=="Confidence Boost")), tracker, track_row)
                 with gr.Row(visible=False) as track_row:
-                    settings.append(gr.Slider(0, 1, value=0.1, label="Low Conf Threshold", info=""))
-                    settings.append(gr.Slider(0, 1, value=0.3, label="High Conf Threshold", info=""))
+                    settings.append(gr.Slider(0, 1, value=default_settings.byte_low_conf, label="Low Conf Threshold", info=""))
+                    settings.append(gr.Slider(0, 1, value=default_settings.byte_high_conf, label="High Conf Threshold", info=""))
                     tracker.change(lambda x:  gr.update(visible=(x=="ByteTrack")), tracker, track_row)
 
                 gr.Markdown("Other")
                 with gr.Row():
-                    settings.append(gr.Slider(0, 3, value=0.3, label="Min Length", info="Minimum length of fish (meters) in order for it to count"))
-                    settings.append(gr.Slider(0, 5, value=1, label="Min Travel", info="Minimum travel distance of track (meters) in order for it to count"))
+                    settings.append(gr.Slider(0, 3, value=default_settings.min_length, label="Min Length", info="Minimum length of fish (meters) in order for it to count"))
+                    settings.append(gr.Slider(0, 5, value=default_settings.min_travel, label="Min Travel", info="Minimum travel distance of track (meters) in order for it to count"))
 
                 gradio_components['hyperparams'] = settings