用python和opencv实现图像拼接

Question

有两张图片，它们的面积相同，如：

我想拼接两个图像。我的代码如下

import numpy as np
import cv2

leftImg = cv2.imread('D:\\1.jpg')
rightImg = cv2.imread('D:\\2.jpg')
leftgray=cv2.cvtColor(leftImg,cv2.COLOR_BGR2GRAY)
rightgray=cv2.cvtColor(rightImg,cv2.COLOR_BGR2GRAY)

hessian=400
surf=cv2.SURF(hessian)
kp1,des1=surf.detectAndCompute(leftgray,None)
kp2,des2=surf.detectAndCompute(rightgray,None)


FLANN_INDEX_KDTREE=0
indexParams=dict(algorithm=FLANN_INDEX_KDTREE,trees=5)
searchParams=dict(checks=50)
flann=cv2.FlannBasedMatcher(indexParams,searchParams)
matches=flann.knnMatch(des1,des2,k=2)

h,w=leftgray.shape[:2]
good=[]
for m,n in matches:
    if m.distance < 0.7*n.distance:
        good.append(m)

src_pts = np.array([ kp1[m.queryIdx].pt for m in good])
dst_pts = np.array([ kp2[m.trainIdx].pt for m in good])
H=cv2.findHomography(src_pts,dst_pts)
dst_corners=cv2.warpPerspective(leftgray,H,(w*2,h))

dst_corners[0:h,w:w*2]=rightgray
cv2.imwrite('tiled.jpg',dst_corners)
cv2.imshow('tiledImg',dst_corners)

cv2.waitKey()
cv2.destroyAllWindows()

但我有个错误，那就是

^{pr2}$

我该怎么办？请告诉我如何实现这个功能，非常感谢！在

Answer 1

如果你打印单应矩阵，你就会明白你的错误。把这行改成。在

dst_corners=cv2.warpPerspective(leftgray,H[0],(w*2,h))

Answer 2

按照this solution中的建议，您应该在

H, mask =cv2.findHomography(src_pts,dst_pts)

分离输出