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;我把三个通道的值平均了的结果
