Как вы получаете пары координат из cv2.knnMatch? - PullRequest
0 голосов
/ 13 июня 2019

Я пытаюсь заставить OpenCV (Python) дать мне пары координат (в виде массивов NumPy) совпадающих точек привязки для выравнивания 2 изображений (с аффинным преобразованием, если необходимо).

Я пытался расшифровать "совпадения" объекта, но не могу понять, как извлечь из него координаты пикселей. Я хочу, чтобы пары координат совпадающих точек появлялись в виде массивов NumPy.

import numpy as np
import cv2
from matplotlib import pyplot as plt

MIN_MATCH_COUNT = 10

img1 = cv2.imread('box.png',0)          # queryImage
img2 = cv2.imread('box_in_scene.png',0) # trainImage

# Initiate SIFT detector
sift = cv.xfeatures2d.SIFT_create()

# find the keypoints and descriptors with SIFT
kp1, des1 = sift.detectAndCompute(img1,None)
kp2, des2 = sift.detectAndCompute(img2,None)

FLANN_INDEX_KDTREE = 0
index_params = dict(algorithm = FLANN_INDEX_KDTREE, trees = 5)
search_params = dict(checks = 50)

flann = cv2.FlannBasedMatcher(index_params, search_params)

matches = flann.knnMatch(des1,des2,k=2)
matchesMask = [[0,0] for i in range(len(matches))]
# ratio test as per Lowe's paper
for i,(m,n) in enumerate(matches):
    if m.distance < 0.2*n.distance:
        matchesMask[i]=[1,0]
draw_params = dict(matchColor = (0,255,0),
                   singlePointColor = (255,0,0),
                   matchesMask = matchesMask,
                   flags = 0)
img3 = cv.drawMatchesKnn(img1,kp1,img2,kp2,matches,None,**draw_params)
plt.figure(1234)
plt.imshow(img3)

Я получаю изображение, показывающее совпадающие точки, но я не могу извлечь необработанные пары координат, использованные для создания изображения.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...