update wuxi demo

2026-04-11 12:20:16 +08:00
parent ea21302d7f
commit 92d68cc80e
10 changed files with 696 additions and 13 deletions
@@ -85,11 +85,11 @@ class Tracker:
    def track(self, frame):
        """Main tracking function that dispatches to CUDA or CPU implementation"""
        if self.__algorithm == "raft":
-            return self.track_raft(frame)
+            return self.track_raft(frame.copy())
        elif self.__algorithm == "dis":
            if frame.ndim == 3:
                frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
-            return self.track_cpu(frame)
+            return self.track_cpu(frame.copy())
    
    def track_cpu(self, frame):
        """CPU-based optical flow tracking"""