New Default Config

InferenceConfig.py

@@ -1,22 +1,23 @@
 from enum import Enum
 
+class TrackerType(Enum):
+    NONE = 0
+    CONF_BOOST = 1
+    BYTETRACK = 2
+
 ### Configuration options
 WEIGHTS = 'models/v5m_896_300best.pt'
 # will need to configure these based on GPU hardware
 BATCH_SIZE = 32
 
 CONF_THRES = 0.05 # detection
-NMS_IOU  = 0.2 # NMS IOU
-MAX_AGE = 14 # time until missing fish get's new id
-MIN_HITS = 16 # minimum number of frames with a specific fish for it to count
+NMS_IOU  = 0.25 # NMS IOU
+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
-
-class TrackerType(Enum):
-    NONE = 0
-    CONF_BOOST = 1
-    BYTETRACK = 2
+DEFAULT_TRACKER = TrackerType.BYTETRACK
 
 class InferenceConfig:
     def __init__(self, 
@@ -30,11 +31,14 @@ class InferenceConfig:
         self.min_length = min_length
         self.min_travel = min_travel
 
+        self.associative_tracker = DEFAULT_TRACKER
+        self.boost_power = 2
+        self.boost_decay = 0.1
+        self.byte_low_conf = 0.1
+        self.byte_high_conf = 0.3
+
+    def enable_sort_track(self):
         self.associative_tracker = TrackerType.NONE
-        self.boost_power = 1
-        self.boost_decay = 1
-        self.byte_low_conf = 1
-        self.byte_high_conf = 1
 
     def enable_conf_boost(self, power, decay):
         self.associative_tracker = TrackerType.CONF_BOOST

app.py

@@ -56,6 +56,8 @@ def on_aris_input(file_list, model_id, conf_thresh, iou_thresh, min_hits, max_ag
         state['config'].enable_conf_boost(boost_power, boost_decay)
     elif (associative_tracker == "ByteTrack"):
         state['config'].enable_byte_track(byte_low_conf, byte_high_conf)
+    else: 
+        state['config'].enable_sort_track()
 
     print(" ")
     print("Running with:")