Python Opencv实践 - 全景图片拼接stitcher

news2024/9/21 11:04:49

做一个全景图片切片的程序Spliter

由于手里没有切割好的全景图片资源，因此首先写了一个切片的程序spliter。

如果有现成的切割好的待拼接的切片文件，则不需要使用spliter。

对于全景图片的拼接，需要注意一点，各个切片图片之间要有重复的内容以便opencv能够提取到关键点并能匹配上。如果简单地将全景图均分几乎是不可能成功的，stitcher一般会返回错误1（ERR_NEED_MORE_IMGS）。下面是我写的切片程序spliter代码，仅供参考：

import cv2 as cv
import numpy as np
import matplotlib.pyplot as plt

img = cv.imread('../../SampleImages/panoramaOriginal.jpg')
numSlices = 5
print(img.shape)
(height,width,channels) = img.shape

sliceWidth = width / numSlices
#全景拼接需要匹配两幅图中的特征点，因此需要预留一部分重复的列
#sliceDuplicateOffset设置了两幅图重复的列的数量
sliceDuplicateOffset = sliceWidth / 2
print(sliceWidth)
sliceStartX = 0
for i in range(0, numSlices):
    sliceX = sliceStartX;
    #第一个切片不需要减去sliceDuplicateOffset
    if (sliceStartX != 0):
        sliceX = sliceX - sliceDuplicateOffset
    imgSlice = img[0:height,int(sliceX):int(sliceStartX + sliceWidth)]
    sliceStartX += sliceWidth
    cv.imwrite('panoramaSlice' + str(i) + ".jpg", imgSlice, [cv.IMWRITE_JPEG_QUALITY,100])

我下载的原始全景图片为：

经过spliter切片后

读取切片文件使用Stitcher进行拼接

将切片文件全部放到相同目录下，通过os库进行读取放到一个list中，然后使用Stitcher进行拼接，代码如下：

import cv2 as cv
import numpy as np
import matplotlib.pyplot as plt
import os

slices = []
imageDir = '../../SampleImages/panoramaSlices/'
imageFiles = os.listdir(imageDir)
#读取所有切片文件
for file in imageFiles: 
    img = cv.imread(imageDir+file)
    slices.append(img)

#plt.imshow(slices[0])

#创建stitcher对象
#cv.Stitcher.Create()
#参考资料：https://blog.csdn.net/qq_41112170/article/details/124634544
stitcher = cv.Stitcher.create()
#调用stitch做全景拼接
#参考资料：https://blog.csdn.net/weixin_48911487/article/details/122721333
#         https://www.cnblogs.com/rainsoul/p/8430074.html
(status,result) = stitcher.stitch(slices)
if status == cv.STITCHER_OK:
    print("Panorama stitched successfully!")
    plt.imshow(result[:,:,::-1])
else:
    print("Panorama stitch failed! Status=" + str(status))

运行效果：