fix dtype bug

mikel-brostrom · Jun 20, 2023 · 4c294f9 · 4c294f9
1 parent a504e9e
commit 4c294f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/boxmot/strongsort/sort/track.py b/boxmot/strongsort/sort/track.py
@@ -276,7 +276,7 @@ def update(self, detection, class_id, conf):
             The associated detection.
         """
         self.conf = conf
-        self.class_id = class_id.int()
+        self.class_id = class_id.astype('int64')
         self.mean, self.covariance = self.kf.update(self.mean, self.covariance, detection.to_xyah(), detection.confidence)
 
         feature = detection.feature / np.linalg.norm(detection.feature)