fixed a bug (#181)

Co-authored-by: lihi <[email protected]>
cheind · Sep 28, 2023 · 22483cb · 22483cb
1 parent 0708b3f
commit 22483cb
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/motmetrics/mot.py b/motmetrics/mot.py
@@ -257,9 +257,19 @@ def update(self, oids, hids, dists, frameid=None, vf=''):
 
                 o = oids[i]
                 h = hids[j]
-                is_switch = (o in self.m and
-                             self.m[o] != h and
-                             abs(frameid - self.last_occurrence[o]) <= self.max_switch_time)
+                ######################################################################
+                # todo - fixed a bug:
+                # is_switch = (o in self.m and
+                #              self.m[o] != h and
+                #              abs(frameid - self.last_occurrence[o]) <= self.max_switch_time)
+                switch_condition = (
+                        o in self.m and
+                        self.m[o] != h and
+                        o in self.last_occurrence and  # Ensure the object ID 'o' is initialized in last_occurrence
+                        abs(frameid - self.last_occurrence[o]) <= self.max_switch_time
+                )
+                is_switch = switch_condition
+                ######################################################################
                 cat1 = 'SWITCH' if is_switch else 'MATCH'
                 if cat1 == 'SWITCH':
                     if h not in self.hypHistory: