OpenCV 实现SSIM结构相似性算法

SSIM算法的介绍:
http://blog.csdn.net/chaipp0607/article/details/70158835

代码做了一下处理:
(1)设置两组对比试验,将原图进行核为55的滤波,与原图比较求得SSIM指数。将原图进行核为1010的滤波,与原图比较求得SSIM指数。
(2)将SSIM指数折算为百分制
(3)采用高斯模糊求得图像的均值

代码参考:
http://jingyan.baidu.com/article/456c463b67aa310a5931447a.html

#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <iostream>

using namespace std; 
using namespace cv;

Scalar getMSSIM(Mat  inputimage1, Mat inputimage2);
int main()
{
	Mat BlurImage1;
	Mat BlurImage2;
	Mat SrcImage = imread("1.jpg");
	blur(SrcImage, BlurImage1, Size(5, 5));
	blur(SrcImage,BlurImage2,Size(10,10));
	Scalar SSIM1 = getMSSIM(SrcImage, BlurImage1);
	Scalar SSIM2 = getMSSIM(SrcImage, BlurImage2);
	printf("模糊5*5通道1:%f\n", SSIM1.val[0] * 100);
	printf("模糊5*5通道2:%f\n", SSIM1.val[1] * 100);
	printf("模糊5*5通道3:%f\n", SSIM1.val[2] * 100);
	printf("模糊5*5:%f\n", (SSIM1.val[2] + SSIM1.val[1] + SSIM1.val[0])/3 * 100);
	printf("模糊10*10通道1:%f\n", SSIM2.val[0] * 100);
	printf("模糊10*10通道2:%f\n", SSIM2.val[1] * 100);
	printf("模糊10*10通道3:%f\n", SSIM2.val[2] * 100);
	printf("模糊10*10:%f\n", (SSIM2.val[2] + SSIM2.val[1] + SSIM2.val[0]) / 3 * 100);
	imshow("原图",SrcImage);
	imshow("模糊5*5",BlurImage1);
	imshow("模糊10*10", BlurImage2);
	waitKey(0);
	return 0;
}
Scalar getMSSIM(Mat  inputimage1, Mat 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);
	i2.convertTo(I2, d);
	Mat I2_2 = I2.mul(I2);
	Mat I1_2 = I1.mul(I1);
	Mat I1_I2 = I1.mul(I2);
	Mat mu1, mu2;
	GaussianBlur(I1, mu1, Size(11, 11), 1.5);
	GaussianBlur(I2, mu2, Size(11, 11), 1.5);
	Mat mu1_2 = mu1.mul(mu1);
	Mat mu2_2 = mu2.mul(mu2);
	Mat mu1_mu2 = mu1.mul(mu2);
	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;
}

打印结果:
模糊55通道1:82.523627
模糊5
5通道2:85.781376
模糊55通道3:85.903646
模糊5
5:84.736216
模糊1010通道1:65.029142
模糊10
10通道2:69.286267
模糊1010通道3:68.664205
模糊10
10:67.659871

这里写图片描述
这里写图片描述
这里写图片描述

SSIM算法是一种用于测量两张图像之间结构相似性算法,可以用来评估图像质量。下面是一个简单的C++实现: ```c++ #include <opencv2/opencv.hpp> #include <iostream> using namespace cv; using namespace std; double getSSIM(Mat& img1, Mat& img2) { const double C1 = 6.5025, C2 = 58.5225; int d = CV_32F; Mat I1, I2; img1.convertTo(I1, d); img2.convertTo(I2, d); Mat I2_2 = I2.mul(I2); Mat I1_2 = I1.mul(I1); Mat I1_I2 = I1.mul(I2); Mat mu1, mu2; GaussianBlur(I1, mu1, Size(11, 11), 1.5); GaussianBlur(I2, mu2, Size(11, 11), 1.5); Mat mu1_2 = mu1.mul(mu1); Mat mu2_2 = mu2.mul(mu2); Mat mu1_mu2 = mu1.mul(mu2); 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[0]; } int main() { Mat img1 = imread("img1.jpg", IMREAD_GRAYSCALE); Mat img2 = imread("img2.jpg", IMREAD_GRAYSCALE); double ssim = getSSIM(img1, img2); cout << "SSIM: " << ssim << endl; return 0; } ``` 其中,`getSSIM()`函数用于计算两张图像的SSIM值。具体实现是先对图像进行高斯模糊,然后计算图像均值、方差和协方差等参数,并根据这些参数计算出SSIM值。在`main()`函数中,我们读取两张图像并调用`getSSIM()`函数来计算它们的SSIM值。 需要注意的是,这里使用了OpenCV库来读取图像和进行高斯模糊等操作,因此需要安装OpenCV库并将其链接到程序中。
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值