mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-06 14:36:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #3628 from cbenhagen:patch-2

Browse Source
This commit is contained in:
Vadim Pisarevsky 2015-01-20 12:18:04 +00:00
commit ea38f7fa66
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
samples/python2/plane_tracker.py
View File

@ -87,10 +87,10 @@ class PlaneTracker:
def track(self, frame): def track(self, frame):
'''Returns a list of detected TrackedTarget objects''' '''Returns a list of detected TrackedTarget objects'''
self.frame_points, self.frame_descrs = self.detect_features(frame) frame_points, frame_descrs = self.detect_features(frame)
if len(self.frame_points) < MIN_MATCH_COUNT: if len(frame_points) < MIN_MATCH_COUNT:
return [] return []
matches = self.matcher.knnMatch(self.frame_descrs, k = 2) matches = self.matcher.knnMatch(frame_descrs, k = 2)
matches = [m[0] for m in matches if len(m) == 2 and m[0].distance < m[1].distance * 0.75] matches = [m[0] for m in matches if len(m) == 2 and m[0].distance < m[1].distance * 0.75]
if len(matches) < MIN_MATCH_COUNT: if len(matches) < MIN_MATCH_COUNT:
return [] return []
@ -103,7 +103,7 @@ class PlaneTracker:
continue continue
target = self.targets[imgIdx] target = self.targets[imgIdx]
p0 = [target.keypoints[m.trainIdx].pt for m in matches] p0 = [target.keypoints[m.trainIdx].pt for m in matches]
p1 = [self.frame_points[m.queryIdx].pt for m in matches] p1 = [frame_points[m.queryIdx].pt for m in matches]
p0, p1 = np.float32((p0, p1)) p0, p1 = np.float32((p0, p1))
H, status = cv2.findHomography(p0, p1, cv2.RANSAC, 3.0) H, status = cv2.findHomography(p0, p1, cv2.RANSAC, 3.0)
status = status.ravel() != 0 status = status.ravel() != 0