利用OpenCV計算灰度圖像的峰值信噪比（PSNR）

我用兩種方法實現了灰度圖像的PSNR的計算，如下。代碼雖說是針對灰度圖像的，但能很容易擴展到多通道圖像的PSNR的計算。

代碼之一,完全靠自己的代碼實現PSNR的計算：

1. #include <cv.h>
2. #include <highgui.h>
3. #include <math.h>
4. #include <stdio.h>
5. #pragma comment( lib, "cv.lib" )
6. #pragma comment( lib, "cxcore.lib" )
7. #pragma comment( lib, "highgui.lib" )
8. int main()
9. {
10. IplImage* img = cvLoadImage("lena.jpg",0);
11. int height=img->height;
12. int width=img->width;
13. int step=img->widthStep;
14. uchar *data=(uchar *)img->imageData;
15. int i,j;
16. double sum=0;
17. for(i=0;i<height;i++)
18. {
19. for(j=0;j<width;j++)
20. {
21. sum+=data[i*step+j];
22. }
23. }
24. double mean=0;
25. mean=sum/(width*height);
26. double mse=0;
27. for(i=0;i<height;i++)
28. {
29. for(j=0;j<width;j++)
30. {
31. mse+=(data[i*step+j]-mean)*(data[i*step+j]-mean);
32. }
33. }
34. mse=mse/(width*height);
35. double psnr=0;
36. psnr=10*log10(255*255/mse);
37. printf("%lf/n",sum);
38. printf("%lf/n",mean);
39. printf("%lf/n",mse);
40. printf("%lf/n",psnr);
41. cvNamedWindow("Lena", CV_WINDOW_AUTOSIZE);
42. cvShowImage("Lena", img );
43. cvWaitKey(0);
44. cvDestroyWindow("Lena");
45. cvReleaseImage(&img );
46. return 0;
47. }

代碼之二，使用了許多OpenCV庫提供的函數，代碼要精煉一些：

1. #include <cv.h>
2. #include <highgui.h>
3. #include <math.h>
4. #include <stdio.h>
5. #pragma comment( lib, "cv.lib" )
6. #pragma comment( lib, "cxcore.lib" )
7. #pragma comment( lib, "highgui.lib" )
8. int main()
9. {
10. IplImage* img = cvLoadImage("lena.jpg",0);
12. CvScalar sum=cvSum(img);
13. CvScalar mean=cvAvg(img);
14. CvScalar stddev;
15. cvAvgSdv(img,NULL,&stddev);
16. double psnr=20*log10(255/stddev.val[0]);
17. printf("%lf/n",sum.val[0]);
18. printf("%lf/n",mean.val[0]);
19. printf("%lf/n",stddev.val[0]);
20. printf("%lf/n",psnr);
21. cvNamedWindow("Lena", CV_WINDOW_AUTOSIZE);
22. cvShowImage("Lena", img );
23. cvWaitKey(0);
24. cvDestroyWindow("Lena");
25. cvReleaseImage(&img );
26. return 0;
27. }

經lena圖檢驗，上面兩份代碼計算的結果是一樣的。

說明：上面的PSNR的定義參考了維基百科的相關條目。疏漏之處，歡迎拍磚！