多波段融合算法C++代码

这里提供一种基于多波段融合的算法的C++代码实现，供您参考：

#include <iostream>
#include <opencv2/opencv.hpp>

#define IMG_NUM 3       // 融合图像数量
#define WAVE_NUM 3      // 小波分解层数

using namespace std;
using namespace cv;

void wavelet_fusion(vector<Mat>& src, Mat& dst) {
    // 多波段融合算法
    vector<Mat> wavelet_src[IMG_NUM];       // 存储每个图像的小波分解系数
    vector<Mat> fusion_coeff[WAVE_NUM+1];   // 存储融合后的小波分解系数

    // 对每个图像进行小波分解
    for (int i = 0; i < IMG_NUM; i++) {
        for (int j = 0; j <= WAVE_NUM; j++) {
            Mat coeff, temp;
            pyrDown(src[i], temp, Size(src[i].cols/2, src[i].rows/2));
            pyrUp(temp, coeff, src[i].size());
            wavelet_src[i].push_back(src[i] - coeff);
            src[i] = temp;
        }
    }

    // 对每个小波分解系数进行融合
    for (int i = 0; i <= WAVE_NUM; i++) {
        Mat fusion = Mat::zeros(wavelet_src[0][i].size(), wavelet_src[0][i].type());
        for (int j = 0; j < IMG_NUM; j++) {
            fusion += wavelet_src[j][i];
        }
        fusion /= IMG_NUM;
        fusion_coeff[i].push_back(fusion);
    }

    // 对融合后的小波分解系数进行重构
    for (int i = WAVE_NUM; i >= 0; i--) {
        Mat temp = fusion_coeff[i].back();
        for (int j = 0; j < IMG_NUM; j++) {
            Mat expand;
            pyrUp(temp, expand, Size(wavelet_src[j][i].cols, wavelet_src[j][i].rows));
            temp = expand + wavelet_src[j][i];
        }
        fusion_coeff[i].push_back(temp);
    }

    // 融合后的图像为重构后的第一层小波分解系数
    dst = fusion_coeff[0].back();
}

int main() {
    // 读取图像
    vector<Mat> src(IMG_NUM);
    for (int i = 0; i < IMG_NUM; i++) {
        string filename = "image" + to_string(i+1) + ".jpg";
        src[i] = imread(filename);
        if (src[i].empty()) {
            cout << "Can't read image " << filename << endl;
            return -1;
        }
    }

    // 多波段融合
    Mat dst;
    wavelet_fusion(src, dst);

    // 显示结果
    namedWindow("input", WINDOW_NORMAL);
    for (int i = 0; i < IMG_NUM; i++) {
        string winname = "image" + to_string(i+1);
        namedWindow(winname, WINDOW_NORMAL);
        imshow(winname, src[i]);
    }
    namedWindow("output", WINDOW_NORMAL);
    imshow("output", dst);
    waitKey(0);

    return 0;
}

注意，这里使用的是OpenCV库中的小波变换函数`pyrDown`和`pyrUp`，需要在编译时链接OpenCV库。同时，代码中读取的图像数量为3，需要根据实际情况修改。