c++配置opencv在这里:https://www.jianshu.com/p/5518046a5797
代码:
#include <iostream>
#include <opencv2\core\core.hpp>
#include <opencv2\highgui\highgui.hpp>
#include <opencv2\imgproc\imgproc.hpp>
#include <cmath>
using namespace cv;
Scalar getMSSIM(Mat inputimage1, Mat inputimage2)
{
//x:inputimage1 y:inputimage2
Mat i1 = inputimage1;
Mat i2 = inputimage2;
const double C1 = 6.5025, C2 = 58.5225;
int d = CV_32F;
Mat I1, I2;
i1.convertTo(I1, d);//将CV_8U转成CV_32F
i2.convertTo(I2, d);
Mat I2_2 = I2.mul(I2);//y^2
Mat I1_2 = I1.mul(I1);//x^2
Mat I1_I2 = I1.mul(I2);//x*y
Mat mu1, mu2;
GaussianBlur(I1, mu1, Size(11, 11), 1.5); //u_x
GaussianBlur(I2, mu2, Size(11, 11), 1.5); //u_y
Mat mu1_2 = mu1.mul(mu1); //u_x^2
Mat mu2_2 = mu2.mul(mu2); //u_y^2
Mat mu1_mu2 = mu1.mul(mu2); //u_y*u_x
Mat sigma1_2, sigma2_2, sigma12;
GaussianBlur(I1_2, sigma1_2, Size(11, 11), 1.5);
sigma1_2 -= mu1_2;
GaussianBlur(I2_2, sigma2_2, Size(11, 11), 1.5);
sigma2_2 -= mu2_2;
GaussianBlur(I1_I2, sigma12, Size(11, 11), 1.5);
sigma12 -= mu1_mu2;
Mat t1, t2, t3;
t1 = 2 * mu1_mu2 + C1;
t2 = 2 * sigma12 + C2;
t3 = t1.mul(t2);
t1 = mu1_2 + mu2_2 + C1;
t2 = sigma1_2 + sigma2_2 + C2;
t1 = t1.mul(t2);
Mat ssim_map;
divide(t3, t1, ssim_map);
Scalar mssim = mean(ssim_map);
return mssim;
}
Mat img2 = imread("D:\\project_hdr\\myGan_same_size_wights_noexpand\\test_data_hdr\\5892_256x256.hdr");
Mat img2_p_706 = imread("D:\\project_hdr\\myGan_same_size_wights_noexpand\\test_data_ldr\\706\\5892_256x256_prediction.hdr");
Scalar msg2_706 = getMSSIM(img2, img2_p_706);
std::cout << getAVER(msg2_706) << std::endl;
我把三个通道的值平均了的结果