【opencv】图片拼接

python实现

import cv2
import os

mainFolder = 'Images'#总文件夹目录，含子文件夹
myFolders = os.listdir(mainFolder)#os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表,注意返回值为列表
print(myFolders)#打印指定的文件夹包含的文件或文件夹的名字的列表

for folder in myFolders:#循环总文件夹目录
    path = mainFolder +'/'+folder#folder为数字     1  2
    images =[]  #图片列表
    myList = os.listdir(path)
    print(f'Total no of images detected {len(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.imshow(folder,result)#以文件夹为显示名
        cv2.imwrite(f'{folder}.jpg',result)  #保存图片
        cv2.waitKey(1)
    else:
        print('Panorama Generation Unsuccessful')

cv2.waitKey(0)

C++实现

#include <fstream> 
#include "opencv2/stitching/stitcher.hpp"//\opencv\sources\samples\cpp\stitching.cpp 
#include <opencv2/core/core.hpp>  
#include <opencv2/highgui/highgui.hpp>  
#include <iostream> 
  
using namespace cv; 
using namespace std; 
  
vector<Mat> imgs; //保存拼接的原始图像向量 
  
//导入所有原始拼接图像函数 
void parseCmdArgs(int argc, char** argv); 
  
int main(int argc, char* argv[]) 
{ 
  //导入拼接图像 
  parseCmdArgs(argc, argv);   
  Mat pano; 
  Stitcher stitcher = Stitcher::createDefault(false); 
  Stitcher::Status status = stitcher.stitch(imgs, pano);//拼接 
  if (status != Stitcher::OK) //判断拼接是否成功 
  { 
    cout << "Can't stitch images, error code = " << int(status) << endl; 
    return -1; 
  } 
  namedWindow("全景拼接",0); 
  imshow("全景拼接",pano); 
  imwrite("D:\\全景拼接.jpg",pano); 
  waitKey();   
  return 0; 
} 
  
//导入所有原始拼接图像函数 
void parseCmdArgs(int argc, char** argv) 
{ 
  for(int i=1;i<argc;i++) 
  { 
    Mat img = imread(argv[i]); 
    if (img.empty()) 
    { 
      cout << "Can't read image '" << argv[i] << "'\n"; 
    } 
    imgs.push_back(img); 
  } 
}