Min length hyperparam

app.py

@@ -27,7 +27,7 @@ result = {}
 
 
 # Called when an Aris file is uploaded for inference
-def on_aris_input(file_list, model_id, conf_thresh, iou_thresh, min_hits, max_age, use_associative):
+def on_aris_input(file_list, model_id, conf_thresh, iou_thresh, min_hits, max_age, use_associative, min_length):
 
     # Reset Result
     reset_state(result, state)
@@ -40,6 +40,7 @@ def on_aris_input(file_list, model_id, conf_thresh, iou_thresh, min_hits, max_ag
         'min_hits': min_hits,
         'max_age': max_age,
         'use_associative_tracking': use_associative,
+        'min_length': min_length
     }
 
     print(" ")

gradio_scripts/upload_ui.py

@@ -34,6 +34,9 @@ def Upload_Gradio(gradio_components):
                     gr.Markdown("Associative Tracking")
                     settings.append(gr.Checkbox(value=False, label="Enabled"))
 
+                gr.Markdown("Other")
+                settings.append(gr.Slider(0, 3, value=0.3, label="Min Length", info="Minimum length of fish (meters) in order for it to count"))
+
                 gradio_components['hyperparams'] = settings
 
             #Input field for aris submission

inference.py

@@ -58,6 +58,7 @@ def do_full_inference(dataloader, image_meter_width, image_meter_height, gp=None
     if 'max_age' not in hyperparams: hyperparams['max_age'] = MAX_AGE
     if 'use_associative_tracking' not in hyperparams: hyperparams['use_associative_tracking'] = False
     if 'AT_decay' not in hyperparams: hyperparams['AT_decay'] = MIN_HITS
+    if 'min_length' not in hyperparams: hyperparams['min_length'] = MIN_LENGTH
     
     model, device = setup_model(hyperparams['model'])
 
@@ -97,7 +98,7 @@ def do_full_inference(dataloader, image_meter_width, image_meter_height, gp=None
 
     all_preds, real_width, real_height = format_predictions(image_shapes, outputs, width, height, gp=gp)
 
-    results = do_tracking(all_preds, image_meter_width, image_meter_height, min_hits=hyperparams['min_hits'], max_age=hyperparams['max_age'], gp=gp)
+    results = do_tracking(all_preds, image_meter_width, image_meter_height, min_hits=hyperparams['min_hits'], max_age=hyperparams['max_age'], min_length=hyperparams['min_length'], gp=gp)
 
     return results