• 大小: 2KB
    文件类型: .cpp
    金币: 2
    下载: 1 次
    发布日期: 2021-06-17
  • 语言: C/C++
  • 标签: OPENCV  PSNR  SSIM  

资源简介

使用OPENCV3.0.0 C++写的测试两幅图片之间的对比指标 PSNR SSIM指标

资源截图

代码片段和文件信息

#include
#include
#include
using namespace std;

#include
#include
#include
using namespace cv;

void main()
{
Mat I1 = imread(“D:/results/window-0.jpg“1);
Mat I2 = imread(“D:/results/window-1.jpg“ 1);
double psnrssim;
Mat s1;
absdiff(I1 I2 s1);       // |I1 - I2|AbsDiff函数是 OpenCV 中计算两个数组差的绝对值的函数
s1.convertTo(s1 CV_32F);  // 这里我们使用的CV_32F来计算,因为8位无符号char是不能进行平方计算
s1 = s1.mul(s1);           // |I1 - I2|^2

Scalar s = sum(s1);         //对每一个通道进行加和

double sse = s.val[0] + s.val[1] + s.val[2]; // sum channels

if (sse <= 1e-10) // 对于非常小的值我们将约等于0
psnr = 0;
else
{
double  mse = sse / (double)(I1.channels() * I1.total());//计算MSE
double psnr = 10.0*log10((255 * 255) / mse);
cout << “PSNR=“ << psnr << endl;
cout << “\n“ << endl;
}
I1.convertTo(I1 CV_32F);
I2.convertTo(I2

评论

共有 条评论