OpenCV实战——使用Stitcher进行全景拼接

使用Stitcher进行全景拼接

1. 需求讲解

我们实现一个可以自动扫描文件夹且自动拼接图片的程序。
如我的图片都放在名为Image的文件夹下，且文件夹中有名为1、2的两个文件夹，程序自动扫描文件夹1、2，对其中的图片进行拼接。
注意：如若要拼接成功，对其中的图片要求还是挺高的，不是随便什么图片都能给你拼接出来。

2. 代码实现

2.1 API

os.listdir(path):os.listdir()

f’'：f’’

2.2 完整代码

import cv2
import os

# 这里是为了实现系统自动检索图片文件夹，完成图像拼接
# 将主路径写成我们图片所在的文件夹的名称
mainFolder = 'Image'
myFolders = os.listdir(mainFolder)
# 1.打印我们Image文件夹中所有文件和文件夹的名称，以列表的形式返回
print(myFolders)


for folder in myFolders:
    path = mainFolder + '/' + folder
    images = []
    myList = os.listdir(path)
    # 2.打印文件夹下的图片名称
    print(myList)
    for imgN in myList:
        curImg = cv2.imread(f'{path}/{imgN}')
        curImg = cv2.resize(curImg, (0, 0), None, 0.2, 0.2)
        images.append(curImg)
	
	# 实例化拼接器
    stitcher = cv2.Stitcher.create()
    (status, result) = stitcher.stitch(images)
	# 如果拼接成功就展示图像
    if (status == cv2.STITCHER_OK):
        print("Panorama Generated")
        cv2.imwrite('./Panorama.jpg', result)
        cv2.imshow('Panorama', result)
        cv2.waitKey(1)
    else:
        print('Panorama Generation Unsuccessful')
cv2.waitKey(0)
cv2.destroyAllWindows()

3. 结果展示

拼接成功的图片结果如下：
原图片：


拼接后：