From 6a25fb243d7798ffe80b35e0d0b445e30760435d Mon Sep 17 00:00:00 2001 From: Moses <60121767+mlurbur@users.noreply.github.com> Date: Sat, 30 Mar 2024 14:59:57 -0700 Subject: [PATCH] Fix `None` `prevKeyPoints` for multi-object tracking (#9386) Co-authored-by: UltralyticsAssistant Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com> Co-authored-by: Glenn Jocher --- ultralytics/trackers/utils/gmc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/trackers/utils/gmc.py b/ultralytics/trackers/utils/gmc.py index 806f1b5e..36ea6a2a 100644 --- a/ultralytics/trackers/utils/gmc.py +++ b/ultralytics/trackers/utils/gmc.py @@ -319,7 +319,7 @@ class GMC: keypoints = cv2.goodFeaturesToTrack(frame, mask=None, **self.feature_params) # Handle first frame - if not self.initializedFirstFrame: + if not self.initializedFirstFrame or self.prevKeyPoints is None: self.prevFrame = frame.copy() self.prevKeyPoints = copy.copy(keypoints) self.initializedFirstFrame = True