opencv使用stitching实现全景拼接
原创
©著作权归作者所有:来自51CTO博客作者蒋程扬的部落格的原创作品,请联系作者获取转载授权,否则将追究法律责任
项目中需要使用opencv实现图像的全景拼接,网上找的许多自己实现的算法,最后发现opencv自带的有stitching.hpp可以进行实现,还能使用GPU进行加速,以下是代码:
#include <fstream>
#include "opencv2/highgui/highgui.hpp"
#include "opencv2/stitching.hpp"
#include <iostream>
using namespace cv;
using namespace std;
vector<Mat> imgs; //保存拼接的原始图像向量
//导入所有原始拼接图像函数
void parseCmdArgs(int argc, char** argv);
//这里配置是否使用GPU
bool use_gpu=false;
int main(int argc, char* argv[])
{
//导入拼接图像
parseCmdArgs(argc, argv);
Mat pano;
Stitcher stitcher = Stitcher::createDefault(use_gpu);
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("c:/WorkSpace/opencv/全景拼接.jpg", pano);
waitKey();
return 0;
}
//导入所有原始拼接图像函数
void parseCmdArgs(int argc, char** argv)
{
Mat img;
img = imread("C:/WorkSpace/opencv/inputimg/1.jpg");
imgs.push_back(img);
img = imread("C:/WorkSpace/opencv/inputimg/2.jpg");
imgs.push_back(img);
img = imread("C:/WorkSpace/opencv/inputimg/3.jpg");
imgs.push_back(img);
img = imread("C:/WorkSpace/opencv/inputimg/4.jpg");
imgs.push_back(img);
img = imread("C:/WorkSpace/opencv/inputimg/5.jpg");
imgs.push_back(img);
img = imread("C:/WorkSpace/opencv/inputimg/6.jpg");
imgs.push_back(img);
img = imread("C:/WorkSpace/opencv/inputimg/7.jpg");
imgs.push_back(img);
img = imread("C:/WorkSpace/opencv/inputimg/8.jpg");
imgs.push_back(img);
}
效果:
输入8张图像:
输出效果:
处理的过程会堵塞在这,我的图片是一分钟左右: