feat(ByteTrack): Allow ByteTrack to track detection without class ids

This commit is contained in:
Kader Miyanyedi committed 2024-10-23 22:04:56 +03:00
1 parent 98aedd7383
commit dbd4aa7798
1 file changed
+6 -1
+6 -1
View File
@@ -276,11 +276,16 @@ class ByteTrack:
```
"""
num_rows = detections.xyxy.shape[0]
class_ids = np.full(num_rows, -5)
if detections.class_id is not None:
class_ids = detections.class_id
tensors = np.hstack(
(
detections.xyxy,
detections.confidence[:, np.newaxis],
detections.class_id[:, np.newaxis],
class_ids[:, np.newaxis],
)
)
tracks = self.update_with_tensors(tensors=tensors)