c++与opencv实现mssim

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;

我把三个通道的值平均了的结果
在这里插入图片描述

  • 3
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值