orisys-tech/orisys-sdk
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ensure track cpu use gray image

Browse Source
This commit is contained in:
bluepulsar
2026-01-31 12:26:11 +08:00
parent e2100204aa
commit 2ebbe1a284
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/orisys/tracker.py
+2
View File
@@ -59,6 +59,8 @@ class Tracker:
if self.cuda: if self.cuda:
return self.track_cuda(frame) return self.track_cuda(frame)
else: else:
if frame.ndim == 3:
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
return self.track_cpu(frame) return self.track_cpu(frame)
def track_cuda(self, frame): def track_cuda(self, frame):