ultralytics 8.1.40 search in Python sets {} for speed (#9450)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-04-01 00:16:52 +02:00 · 2024-04-01 00:16:52 +02:00 · ea527507fe
commit ea527507fe
parent 30484d5925
41 changed files with 97 additions and 93 deletions
--- a/ultralytics/trackers/byte_tracker.py
+++ b/ultralytics/trackers/byte_tracker.py
@ -47,7 +47,7 @@ class STrack(BaseTrack):
        """Initialize new STrack instance."""
        super().__init__()
        # xywh+idx or xywha+idx
-        assert len(xywh) in [5, 6], f"expected 5 or 6 values but got {len(xywh)}"
+        assert len(xywh) in {5, 6}, f"expected 5 or 6 values but got {len(xywh)}"
        self._tlwh = np.asarray(xywh2ltwh(xywh[:4]), dtype=np.float32)
        self.kalman_filter = None
        self.mean, self.covariance = None, None