python 特徴量マッチングde

Question

PythonのOpenCVでインターネットのサイトから、
下記のような、特徴量マッチングのプログラム実行しました。
そしたら、
error: OpenCV(4.1.0) C:\projects\opencv-python\opencv\modules\core\src\batch_distance.cpp:303: error: (-215:Assertion failed) K == 1 && update == 0 && mask.empty() in function 'cv::batchDistance'

というエラーが発生しました。

matches・・・ のところでエラーが発生しているらしいのですが、
原因、対応について教えて頂けますでしょうか？

以上です。

img1 = cv2.imread("src/star_ptn.png")
img2 = cv2.imread("src/star.png")
detector1 = cv2.ORB_create()
kps1,des1 = detector1.detectAndCompute(img1,None)
kps2,des2 = detector1.detectAndCompute(img2,None)

bf = cv2.BFMatcher(cv2.NORM_HAMMING,crossCheck = True)
good = []
for m,n in matches:
    if m.distance < 0.5 * n.distacne:
        good.append([m])

img3 = cv2.drawMatchesKnn(img1,kps1,img2,kps2,good,None,flags = 2)
cv2.imshow("kp1",img3)
cv2.waitKey(0)
cv2.destroyAllWindows()

Comments

[amahara_waya]

そもそもmatchesの定義がコード内にありませんが、どこか別の場所で定義してはいないでしょうか？

[soepu]

すいません。記入し忘れました。
bf = ・・・ の次の行に
matches = bf.knnMatch(des1, des2, k=2)
がはいります。

[tiitoi]

以下の点を確認してみてください。
1. img1、img2 が None でないかどうか: None の場合、imread() での画像読み込みに失敗しています。
2. kps1, kps2 がともに空でないかどうか: 空の場合、特徴点マッチングに失敗しています。その場合、今回のケースでは特徴点マッチングは適していないということになります。

[soepu]

print で1、２ともに確認したところ、両方ともNoneではありませんでした。

Answer 1

原因、対応について教えて頂けますでしょうか？

crosscheck は「画像1の i 番目の特徴点が画像2の j 番目の特徴点と最も距離が近い」、かつ「画像2の j 番目の特徴点が画像1の i 番目の特徴点と最も距離が近い」という条件でマッチングするように設定するフラグですが、knnmatch() は1対1対応でなく、1対k対応なので、そのような処理が定義できないのでエラーになる。

対応は、「crosscheck=True の場合、match() を使う」または「crosscheck=Falseにする」

knnmatch() でマッチングする場合

knnmatch() でマッチングする場合、画像1の各特徴点と距離が近い上位 k 個の画像2の特徴点を返します。

import cv2

img1 = cv2.imread('1.png')
img2 = cv2.imread('2.png')

akaze = cv2.ORB_create()
kp1, des1 = akaze.detectAndCompute(img1, None)
kp2, des2 = akaze.detectAndCompute(img2, None)

bf = cv2.BFMatcher()
matches = bf.knnMatch(des1, des2, k=2)

# ratio test
ratio = 0.5
good = []
for m, n in matches:
    if m.distance < ratio * n.distance:
        good.append([m])

# 対応する特徴点同士を描画
img3 = cv2.drawMatchesKnn(img1, kp1, img2, kp2, good, None, flags=2)

cv2.imwrite('img.png', img3)

match() でマッチングする場合

match() でマッチングする場合、最も距離が近い特徴点同士をマッチングします。

import cv2

img1 = cv2.imread('1.png')
img2 = cv2.imread('2.png')

akaze = cv2.ORB_create()
kp1, des1 = akaze.detectAndCompute(img1, None)
kp2, des2 = akaze.detectAndCompute(img2, None)

bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)
matches = bf.match(des1, des2)

# 対応する特徴点同士を描画
img3 = cv2.drawMatches(img1, kp1, img2, kp2, matches, None, flags=2)

cv2.imwrite('img.png', img3)