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

做一个全景图片切片的程序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))

        运行效果:

        

        如果状态返回失败,请参考代码里的链接看看是什么问题

猜你喜欢

转载自blog.csdn.net/vivo01/article/details/134586119