如何在最佳匹配项上绘制边界框？

Question

我如何使用Python在BF MATCHER中为最佳匹配绘制边界框？

- Ali110

---

请阅读在什么情况下我可以在问题中添加“紧急”或其他类似短语，以便获得更快的答案？ - 总结是这不是一种理想的方式来处理志愿者，并且很可能会反过来阻碍获得答案。请勿在您的问题中添加此内容。 - halfer

Answer 1

以下是一个适当的解决方案概述：

1. 在查询图像（img1）上检测关键点和描述符
2. 在目标图像（img2）上检测关键点和描述符
3. 找到两组描述符之间的最佳匹配项或对应项
4. 使用最佳的10个匹配项来形成变换矩阵
5. 根据变换矩阵转换围绕img1的矩形
6. 添加偏移量以将边界框放置在正确的位置
7. 显示结果图像（如下所示）。

以下是代码：

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

img1 = cv2.imread('box.png', 0)          # query Image
img2 = cv2.imread('box_in_scene.png',0)  # target Image

# Initiate SIFT detector
orb = cv2.ORB_create()

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

# create BFMatcher object
bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)

# Match descriptors.
matches = bf.match(des1,des2)

# Sort them in the order of their distance.
matches = sorted(matches, key = lambda x:x.distance)

good_matches = matches[:10]

src_pts = np.float32([ kp1[m.queryIdx].pt for m in good_matches     ]).reshape(-1,1,2)
dst_pts = np.float32([ kp2[m.trainIdx].pt for m in good_matches ]).reshape(-1,1,2)
M, mask = cv2.findHomography(src_pts, dst_pts, cv2.RANSAC,5.0)
matchesMask = mask.ravel().tolist()
h,w = img1.shape[:2]
pts = np.float32([ [0,0],[0,h-1],[w-1,h-1],[w-1,0] ]).reshape(-1,1,2)

dst = cv2.perspectiveTransform(pts,M)
dst += (w, 0)  # adding offset

draw_params = dict(matchColor = (0,255,0), # draw matches in green color
               singlePointColor = None,
               matchesMask = matchesMask, # draw only inliers
               flags = 2)

img3 = cv2.drawMatches(img1,kp1,img2,kp2,good_matches, None,**draw_params)

# Draw bounding box in Red
img3 = cv2.polylines(img3, [np.int32(dst)], True, (0,0,255),3, cv2.LINE_AA)

cv2.imshow("result", img3)
cv2.waitKey()
# or another option for display output
#plt.imshow(img3, 'result'), plt.show()

- Howard GENG

---

请为您的回答添加更多描述，以帮助人们理解并知道为什么这是一个好的解决方案。仅提供代码不会产生好的答案（值得被投票支持）！ - Brian Tompsett - 汤莱恩

我已经在代码中添加了注释。这不够吗？ - Howard GENG

我不建议在这个问题上再花太多精力 - 这个问题太模糊了，很可能会被关闭。 - halfer

请不要在帖子中抱怨投票 - 投票是匿名的，广泛地说，人们应该被允许按照自己的意愿投票。我们尽量抵制紧急乞求 - 如果您看到一个零努力的问题，要求跳过队列，您可以回答它，但如果您能避免的话，我建议您不要这样做。Stack Overflow存在一个非常棘手的“求助吸血鬼”问题，您收到的负评可能就是出于这个原因。 - halfer

"在匹配项周围绘制边界框"，您是指在绿线周围绘制矩形吗？还是在两组匹配的关键点周围绘制矩形？ - Howard GENG