opencv不同聚焦平面的图片景深合成清晰图片

// jshc.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。 // #include <iostream> #include <windows.h> #include <opencv2/opencv.hpp> #include <opencv2/highgui/highgui.hpp> #include <opencv2/imgproc/imgproc.hpp> using namespace std; using namespace cv; Mat GetPreDealImg(Mat srcImg) { Mat grayImg; if (srcImg.channels() == 3) { cvtColor(srcImg, grayImg, CV_BGR2GRAY); } else { grayImg = srcImg.clone(); } Mat XImg, YImg; Sobel(grayImg, XImg, CV_16S, 1, 0, 3, 1, 0, BORDER_REPLICATE); Sobel(grayImg, YImg, CV_16S, 0, 1, 3, 1, 0, BORDER_REPLICATE); imwrite("XImg.bmp", XImg); imwrite("YImg.bmp", YImg); int width = XImg.cols; int heigth = YImg.rows; Mat outputImg = Mat(Size(XImg.cols, XImg.rows), CV_64F); for (int i = 0; i < heigth; i++) { for (int j = 0; j < width; j++) { int16_t Gx = XImg.at<int16_t>(i, j); int16_t Gy = YImg.at<int16_t>(i, j); double v1 = pow(Gx, 2); double v2 = pow(Gy, 2); double val = sqrt(v1 + v2); outputImg.at<double>(i, j) = val; } } imwrite("outputImg.bmp", outputImg); GaussianBlur(outputImg, outputImg, Size(31, 31), 11); return outputImg; } Mat Id_Img(vector<Mat> coll) { int height = coll[0].rows; int width = coll[0].cols; Mat max_id = Mat(Size(width, height), CV_8U); max_id = 0; Mat heightImg = max_id.clone(); heightImg = 18; for (int num = 1; num < coll.size(); num++) { for (int i = 0; i < height; i++) { for (int j = 0; j < width; j++) { double G1 = coll[0].at<double>(i, j); double G2 = coll[num].at<double>(i, j); if ((G1+2) < G2) { double G_Max = (double)G2; coll[0].at<double>(i, j) = G_Max; max_id.at<uchar>(i, j) = num; heightImg.at<uchar>(i, j) = num*20+10; } } } } imwrite("height.bmp", heightImg); return max_id; } Mat ImgMerge(Mat idImg, vector<Mat> srcImg) { Mat mergeImg = Mat(Size(srcImg[0].cols, srcImg[0].rows), srcImg[0].type()); int heigth = idImg.rows; int width = idImg.cols; for (int index = 0; index < srcImg.size(); index++) { for (int i = 0; i < heigth; i++) { for (int j = 0; j < width; j++) { int val_ID = idImg.at<uchar>(i, j); if (val_ID == index) { mergeImg.at<uchar>(i, j) = srcImg[index].at<uchar>(i, j); } else { continue; } } } } return mergeImg; } int main() { DWORD start, end_time; Mat srcImg_0, srcImg_1, srcImg_2, srcImg_3; vector<Mat> vecSrcImg; srcImg_0 = imread("js01.png", 0); srcImg_1 = imread("js02.png", 0); srcImg_2 = imread("js03.png", 0); vecSrcImg.push_back(srcImg_0); vecSrcImg.push_back(srcImg_1); vecSrcImg.push_back(srcImg_2); Mat grayImg_0, grayImg_1, grayImg_2, grayImg_3; vector<Mat> vecGrayImg; start = GetTickCount(); grayImg_0 = GetPreDealImg(srcImg_0); grayImg_1 = GetPreDealImg(srcImg_1); grayImg_2 = GetPreDealImg(srcImg_2); vecGrayImg.push_back(grayImg_0); vecGrayImg.push_back(grayImg_1); vecGrayImg.push_back(grayImg_2); imwrite("grayImage0.bmp", grayImg_0); imwrite("grayImage1.bmp", grayImg_1); imwrite("grayImage2.bmp", grayImg_2); Mat maxIdImg = Id_Img(vecGrayImg); Mat mergeImg = ImgMerge(maxIdImg, vecSrcImg); end_time = GetTickCount() - start; cout << end_time << endl; imwrite("mergeImage.bmp", mergeImg); }