1. 程式人生 > >影象質量評價相關收錄

影象質量評價相關收錄

2018-12-05 轉自 :https://www.cnblogs.com/ranjiewen/p/6390846.html#undefined 

ssim的計算裡面有高斯模糊

為了快速計算,先對每個小塊進行計算,然後計算所有塊的平均值。

可以參考原始碼實現,而且程式碼實現有近似的在裡面!matlab中中影象PSNR和SSIM的計算

“在實際應用中,可以利用滑動窗將影象分塊,令分塊總數為N,考慮到視窗形狀對分塊的影響,採用高斯加權計算每一視窗的均值、方差以及協方差,然後計算對應塊的結構相似度SSIM,最後將平均值作為兩影象的結構相似性度量,即平均結構相似性MSSIM”

function [mssim, ssim_map] = ssim(img1, img2, K, window, L)

 
 

% ========================================================================
% SSIM Index with automatic downsampling, Version 1.0
% Copyright(c) 2009 Zhou Wang
% All Rights Reserved.
%
% ----------------------------------------------------------------------
% Permission to use, copy, or modify this software and its documentation
% for educational and research purposes only and without fee is hereby
% granted, provided that this copyright notice and the original authors'
% names appear on all copies and supporting documentation. This program
% shall not be used, rewritten, or adapted as the basis of a commercial
% software or hardware product without first obtaining permission of the
% authors. The authors make no representations about the suitability of
% this software for any purpose. It is provided "as is" without express
% or implied warranty.
%----------------------------------------------------------------------
%
% This is an implementation of the algorithm for calculating the
% Structural SIMilarity (SSIM) index between two images
%
% Please refer to the following paper and the website with suggested usage
%
% Z. Wang, A. C. Bovik, H. R. Sheikh, and E. P. Simoncelli, "Image
% quality assessment: From error visibility to structural similarity,"
% IEEE Transactios on Image Processing, vol. 13, no. 4, pp. 600-612,
% Apr. 2004.
%
% http://www.ece.uwaterloo.ca/~z70wang/research/ssim/
%
% Note: This program is different from ssim_index.m, where no automatic
% downsampling is performed. (downsampling was done in the above paper
% and was described as suggested usage in the above website.)
%
% Kindly report any suggestions or corrections to

[email protected]
%
%----------------------------------------------------------------------
%
%Input : (1) img1: the first image being compared
% (2) img2: the second image being compared
% (3) K: constants in the SSIM index formula (see the above
% reference). defualt value: K = [0.01 0.03]
% (4) window: local window for statistics (see the above
% reference). default widnow is Gaussian given by
% window = fspecial('gaussian', 11, 1.5);
% (5) L: dynamic range of the images. default: L = 255
%
%Output: (1) mssim: the mean SSIM index value between 2 images.
% If one of the images being compared is regarded as
% perfect quality, then mssim can be considered as the
% quality measure of the other image.
% If img1 = img2, then mssim = 1.
% (2) ssim_map: the SSIM index map of the test image. The map
% has a smaller size than the input images. The actual size
% depends on the window size and the downsampling factor.
%
%Basic Usage:
% Given 2 test images img1 and img2, whose dynamic range is 0-255
%
% [mssim, ssim_map] = ssim(img1, img2);
%
%Advanced Usage:
% User defined parameters. For example
%
% K = [0.05 0.05];
% window = ones(8);
% L = 100;
% [mssim, ssim_map] = ssim(img1, img2, K, window, L);
%
%Visualize the results:
%
% mssim %Gives the mssim value
% imshow(max(0, ssim_map).^4) %Shows the SSIM index map
%========================================================================

 
 


if (nargin < 2 || nargin > 5)
mssim = -Inf;
ssim_map = -Inf;
return;
end

 
 

if (size(img1) ~= size(img2))
mssim = -Inf;
ssim_map = -Inf;
return;
end

 
 

[M,N] = size(img1);

 
 

if (nargin == 2)
if ((M < 11) || (N < 11)) %影象尺寸大小太小 沒有意義
mssim = -Inf;
ssim_map = -Inf;
return
end
window = fspecial('gaussian', 11, 1.5); %標準偏差為1.5 , 11*11的高斯低通濾波器
K(1) = 0.01; % default settings
K(2) = 0.03; %
L = 255; %
end

 
 

if (nargin == 3)
if ((M < 11) || (N < 11))
mssim = -Inf;
ssim_map = -Inf;
return
end
window = fspecial('gaussian', 11, 1.5);
L = 255;
if (length(K) == 2)
if (K(1) < 0 || K(2) < 0)
mssim = -Inf;
ssim_map = -Inf;
return;
end
else
mssim = -Inf;
ssim_map = -Inf;
return;
end
end

 
 

if (nargin == 4)
[H,W] = size(window);
if ((H*W) < 4 || (H > M) || (W > N))
mssim = -Inf;
ssim_map = -Inf;
return
end
L = 255;
if (length(K) == 2)
if (K(1) < 0 || K(2) < 0)
mssim = -Inf;
ssim_map = -Inf;
return;
end
else
mssim = -Inf;
ssim_map = -Inf;
return;
end
end

 
 

if (nargin == 5)
[H,W] = size(window);
if ((H*W) < 4 || (H > M) || (W > N))
mssim = -Inf;
ssim_map = -Inf;
return
end
if (length(K) == 2)
if (K(1) < 0 || K(2) < 0)
mssim = -Inf;
ssim_map = -Inf;
return;
end
else
mssim = -Inf;
ssim_map = -Inf;
return;
end
end

 
 


img1 = double(img1);
img2 = double(img2);

 
 

% automatic downsampling
f = max(1,round(min(M,N)/256));
%downsampling by f
%use a simple low-pass filter
if(f>1)
lpf = ones(f,f);
lpf = lpf/sum(lpf(:));
img1 = imfilter(img1,lpf,'symmetric','same');
img2 = imfilter(img2,lpf,'symmetric','same');

 
 

img1 = img1(1:f:end,1:f:end);
img2 = img2(1:f:end,1:f:end);
end

 
 

C1 = (K(1)*L)^2;
C2 = (K(2)*L)^2;
window = window/sum(sum(window));

 
 

mu1 = filter2(window, img1, 'valid');
mu2 = filter2(window, img2, 'valid');
mu1_sq = mu1.*mu1;
mu2_sq = mu2.*mu2;
mu1_mu2 = mu1.*mu2;
sigma1_sq = filter2(window, img1.*img1, 'valid') - mu1_sq;
sigma2_sq = filter2(window, img2.*img2, 'valid') - mu2_sq;
sigma12 = filter2(window, img1.*img2, 'valid') - mu1_mu2;

 
 

if (C1 > 0 && C2 > 0)
ssim_map = ((2*mu1_mu2 + C1).*(2*sigma12 + C2))./((mu1_sq + mu2_sq + C1).*(sigma1_sq + sigma2_sq + C2));
else
numerator1 = 2*mu1_mu2 + C1;
numerator2 = 2*sigma12 + C2;
denominator1 = mu1_sq + mu2_sq + C1;
denominator2 = sigma1_sq + sigma2_sq + C2;
ssim_map = ones(size(mu1));
index = (denominator1.*denominator2 > 0);
ssim_map(index) = (numerator1(index).*numerator2(index))./(denominator1(index).*denominator2(index));
index = (denominator1 ~= 0) & (denominator2 == 0);
ssim_map(index) = numerator1(index)./denominator1(index);
end

 
 

mssim = mean2(ssim_map);

 
 

return

 

PSNR , 峰值信噪比

通常用來評價一幅影象壓縮後和原影象相比質量的好壞,當然,壓縮後圖像一定會比原影象質量差的,所以就用這樣一個評價指標來規定標準了。PSNR越高,壓縮後失真越小。這裡主要定義了兩個值,一個是均方差MSE,另一個是峰值信噪比PSNR,公式如下:

PSNR越高,影象和原圖越接近。當然,這都是客觀指標,實際評價還有主觀指標,不過主觀的東西就比較模糊了,每個人感覺都會不同的。最常用的全參考視訊質量評價方法有以下2種:

PSNR(峰值信噪比):用得最多,但是其值不能很好地反映人眼主觀感受。一般取值範圍:20-40.值越大,視訊質量越好。

SSIM(結構相似性):計算稍複雜,其值可以較好地反映人眼主觀感受。一般取值範圍:0-1.值越大,視訊質量越好。

PSNR,SSIM計算有如下工具可選:

MSU Video Quality Measurement Tool:商業軟體,圖形化介面,易上手,使用有限制。

Evalvid中的psnr.exe:開源軟體,命令列介面,使用無限制。推薦,適合批處理。

注:MOS(Mean Opnion Score,平均意見分)是主觀評價實驗之後,得到的主觀分數,取值0-100,值越大,代表主觀感受越好。以下實驗資料來自Live資料庫http://live.ece.utexas.edu/research/Quality/

SSIM引數         一種衡量兩幅影象相似度的新指標,其值越大越好,最大為1,經常用到影象處理中,特別在 影象去噪處理中在影象相似度評價上全面超越SNR(signal to noise ratio)和PSNR(peak signal to noise ratio)。 結構相似性理論認為,自然影象訊號是高度結構化的,即畫素間有很強的相關性,特別是空域中最接近的畫素,這種相關性蘊含著視覺場景中物體結構的重要資訊; HVS的主要功能是從視野中提取結構資訊,可以用對結構資訊的度量作為影象感知質量的近似。結構相似性理論是一種不同於以往模擬HVS低階的組成結構的全新思想,與基於HVS特性的方法相比,最大的區別是自頂向下與自底向上的區別。這一新思想的關鍵是從對感知 誤差度量到對感知結構失真度量的轉變。它沒有試圖通過累加與心理物理學簡單認知模式有關的誤差來估計影象質量,而是直接估計兩個複雜結構訊號的結構改變,從而在某種程度上繞開了自然影象內容複雜性及多通道去相關的問題。作為結構相似性理論的實現,結構相似度指數從影象組成的角度將結構資訊定義為獨立於 亮度對比度的,反映場景中物體結構的屬性,並將失真建模為亮度、對比度和結構三個不同因素的組合。用 均值作為亮度的估計,標準差作為 對比度的估計, 協方差作為結構相似程度的度量。           結構相似性指標(英文: structural similarity index, SSIM index)是一種用以衡量兩張 數位影像相似程度的指標。當兩張影像其中一張為無 失真影像,另一張為 失真後的影像,二者的結構相似性可以看成是失真影像的影像品質衡量指標。相較於傳統所使用的影像品質衡量指標,像是 峰值信噪比(英文: PSNR),結構相似性在影像品質的衡量上更能符合人眼對影像品質的判斷。  

實際使用時,簡化起見,一般會將引數設為\alpha=\beta=\gamma=1C_3=C_2/2,得到:

\text{SSIM}(\mathbf{x},\mathbf{y})=\frac{(2\mu_x\mu_y+C_1)(2\sigma_{xy}+C_2)}{(\mu_x^2+\mu_y^2+C_1)(\sigma_x^2+\sigma_y^2+C_2)}

        在計算兩張影像的結構相似性指標時,會開一個區域性性的視窗,一般為N×N的小區塊,計算出視窗內訊號的結構相似性指標,每次以畫素為單位移動視窗,直到整張影像每個位置的區域性結構相似性指標都計算完畢。將全部的區域性結構相似性指標平均起來即為兩張影像的結構相似性指標。

  • 在實際應用中,可以利用滑動窗將影象分塊,令分塊總數為N,考慮到視窗形狀對分塊的影響,採用高斯加權計算每一視窗的均值、方差以及協方差,然後計算對應塊的結構相似度SSIM,最後將平均值作為兩影象的結構相似性度量,即平均結構相似性MSSIM:

opencv和matlab都有現成的函式可以使用。

參考:影象質量評價--SSIM

         全參考視訊質量評價方法(PSNR,SSIM)以及相關資料庫

 影象質量評價指標

MSE: Mean Squared Error 
均方誤差是指引數估計值與引數真值之差平方的期望值; 
MSE可以評價資料的變化程度,MSE的值越小,說明預測模型描述實驗資料具有更好的精確度。

RMSE 
均方誤差:均方根誤差是均方誤差的算術平方根

MAE :Mean Absolute Error 
平均絕對誤差是絕對誤差的平均值 
平均絕對誤差能更好地反映預測值誤差的實際情況.

SD :standard Deviation 
標準差:標準差是方差的算術平方根。標準差能反映一個數據集的離散程度。平均數相同的兩組組資料,標準差未必相同。

       Spearman秩序相關係數(SROCC)本身就不是衡量線性相關的,而是衡量秩序的相關性的。設有兩組序列X和Y,其秩序為R(X)和R(Y),這裡R(Xi)=k代表Xi是序列X中的第k大(或第k小),則SROCC(X, Y) = PLCC(R(X), R(Y)),其中PLCC是Pearson線性相關係數。
      SROCC被認為是最好的非線性相關指標,這是因為,SROCC只與序列中元素的排序有關。因此即使X或Y被任何單調非線性變換作用(如對數變換、指數變換),都不會對SROCC造成任何影響,因為不會影響元素的排序。維基百科有一張很直觀的圖片,很好地體現了SROCC與PLCC的區別。

 除此之外,還有一個Kendall秩序相關係數(KROCC),它的定義是pairwise的:
KROCC = (同序對數-逆序對數) / 總對數    (總對數=n(n - 1)/2)它的性質與SROCC類似,但是比較抖。

對於連續輸出的機器學習模型或簡單的迴歸擬合,發Paper的話,一般要放SROCC和PLCC,這兩個標準是最重要的,可以很好地衡量模型的非線性相關性和線性相關性。偶爾也會放KROCC和RMSE(方均根誤差)。但是SROCC是比PLCC更為根本的指標,因為一個複雜模型總可以用一些附加的非線性變換(對數或者指數,還有某些領域內recommanded複雜的組合= =)來補正,使得PLCC更高一些,而SROCC卻不能被這些附加非線性變換來影響。