diff options
author | Daniil Kazantsev <dkazanc@hotmail.com> | 2017-07-04 09:35:02 +0100 |
---|---|---|
committer | Daniil Kazantsev <dkazanc@hotmail.com> | 2017-07-04 09:35:02 +0100 |
commit | e097a4edcced2bbc8c78d1302467bdf625deff1d (patch) | |
tree | a848dc49c06ed462988d2f6cce7f2a2212bc4f36 /supp/ssim_index.m | |
parent | 75e7ebc79e86710c4cfb05b8af8e429146288c85 (diff) | |
download | regularization-e097a4edcced2bbc8c78d1302467bdf625deff1d.tar.gz regularization-e097a4edcced2bbc8c78d1302467bdf625deff1d.tar.bz2 regularization-e097a4edcced2bbc8c78d1302467bdf625deff1d.tar.xz regularization-e097a4edcced2bbc8c78d1302467bdf625deff1d.zip |
some clearing
Diffstat (limited to 'supp/ssim_index.m')
-rw-r--r-- | supp/ssim_index.m | 181 |
1 files changed, 0 insertions, 181 deletions
diff --git a/supp/ssim_index.m b/supp/ssim_index.m deleted file mode 100644 index 4fa7a79..0000000 --- a/supp/ssim_index.m +++ /dev/null @@ -1,181 +0,0 @@ -function [mssim, ssim_map] = ssim_index(img1, img2, K, window, L) - -%======================================================================== -%SSIM Index, Version 1.0 -%Copyright(c) 2003 Zhou Wang -%All Rights Reserved. -% -%This is an implementation of the algorithm for calculating the -%Structural SIMilarity (SSIM) index between two images. Please refer -%to the following paper: -% -%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. -% -%Kindly report any suggestions or corrections to zhouwang@ieee.org -% -%---------------------------------------------------------------------- -% -%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: -% size(img1) - size(window) + 1. -% -%Default Usage: -% Given 2 test images img1 and img2, whose dynamic range is 0-255 -% -% [mssim ssim_map] = ssim_index(img1, img2); -% -%Advanced Usage: -% User defined parameters. For example -% -% K = [0.05 0.05]; -% window = ones(8); -% L = 100; -% [mssim ssim_map] = ssim_index(img1, img2, K, window, L); -% -%See the results: -% -% mssim %Gives the mssim value -% imshow(max(0, ssim_map).^4) %Shows the SSIM index map -% -%======================================================================== - - -if (nargin < 2 | nargin > 5) - ssim_index = -Inf; - ssim_map = -Inf; - return; -end - -if (size(img1) ~= size(img2)) - ssim_index = -Inf; - ssim_map = -Inf; - return; -end - -[M N] = size(img1); - -if (nargin == 2) - if ((M < 11) | (N < 11)) % ͼССû塣 - ssim_index = -Inf; - ssim_map = -Inf; - return - end - window = fspecial('gaussian', 11, 1.5); % һƫ1.511*11ĸ˹ͨ˲ - K(1) = 0.01; % default settings - K(2) = 0.03; % - L = 255; % -end - -if (nargin == 3) - if ((M < 11) | (N < 11)) - ssim_index = -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) - ssim_index = -Inf; - ssim_map = -Inf; - return; - end - else - ssim_index = -Inf; - ssim_map = -Inf; - return; - end -end - -if (nargin == 4) - [H W] = size(window); - if ((H*W) < 4 | (H > M) | (W > N)) - ssim_index = -Inf; - ssim_map = -Inf; - return - end - L = 255; - if (length(K) == 2) - if (K(1) < 0 | K(2) < 0) - ssim_index = -Inf; - ssim_map = -Inf; - return; - end - else - ssim_index = -Inf; - ssim_map = -Inf; - return; - end -end - -if (nargin == 5) - [H W] = size(window); - if ((H*W) < 4 | (H > M) | (W > N)) - ssim_index = -Inf; - ssim_map = -Inf; - return - end - if (length(K) == 2) - if (K(1) < 0 | K(2) < 0) - ssim_index = -Inf; - ssim_map = -Inf; - return; - end - else - ssim_index = -Inf; - ssim_map = -Inf; - return; - end -end -%% -C1 = (K(1)*L)^2; % C1Lxyá -C2 = (K(2)*L)^2; % C2ԱȶCxyá -window = window/sum(sum(window)); %˲һ -img1 = double(img1); -img2 = double(img2); - -mu1 = filter2(window, img1, 'valid'); % ͼ˲ӼȨ -mu2 = filter2(window, img2, 'valid'); % ͼ˲ӼȨ - -mu1_sq = mu1.*mu1; % Uxƽֵ -mu2_sq = mu2.*mu2; % Uyƽֵ -mu1_mu2 = mu1.*mu2; % Ux*Uyֵ - -sigma1_sq = filter2(window, img1.*img1, 'valid') - mu1_sq; % sigmax -sigma2_sq = filter2(window, img2.*img2, 'valid') - mu2_sq; % sigmay -sigma12 = filter2(window, img1.*img2, 'valid') - mu1_mu2; % sigmaxy - -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
\ No newline at end of file |