ensure track cpu use gray image

2026-01-31 12:26:11 +08:00
parent e2100204aa
commit 2ebbe1a284
1 changed files with 2 additions and 0 deletions
@@ -59,6 +59,8 @@ class Tracker:
        if self.cuda:
            return self.track_cuda(frame)
        else:
+            if frame.ndim == 3:
+                frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
            return self.track_cpu(frame)
    
    def track_cuda(self, frame):