【图像去噪】全变分算法图像去噪（含SNR）【含Matlab源码 626期】.zip

function [img_estimated,energyi,ISNRi,energyo,ISNRo]=tvmm_debluring(img_noisy,h,lambda,varargin) % function [img_estimated]=tvmm_debluring(img_noisy,h,lamdba,... % 'optional_parameter_name1',value1,'optional_parameter_name2', value2,...); % % Total Variation-based image deconvolution with % a majorization-minimization approach. % % Written by: Joao Oliveira, Jose Bioucas-Dias, Mario Figueiredo % email: joao.oliveira@lx.it.pt % SITE: www.lx.it.pt/~jpaos/tvmm % Date: 21/11/2005 % % % ========================== INPUT PARAMETERS (required) ================= % Parameter Values % name and description % ======================================================================== % img_noisy (double) Noisy blured image of size ny. % h (double) Blur kernel. % lambda Regularization parameter (which is multiplied by the % TV penalty). % % ======================== OPTIONAL INPUT PARAMETERS ==================== % Parameter Values % name and description % ======================================================================= % boa_iter (double) The number of outer loop iterations. % Default: 20 % cg_iter (double) The max number of inner CG iterations. % Default: 200 % soim_iter (double) The max number of inner SOIM iterations. % Default: 20 % soim_cg_iter (double) The max number of CG iterations in inner SOIM. % Default: 6 % cg_thrld (double) Conjugate Gradient threshold. % Default: 1e-5; % image (double) Original image. % displayIm ({'yes','no'}) if 'yes' is passed the restored imaged % is displayed along the CG iterations % Default: 'no' % info_energyi ({'yes','no'}) if 'yes' is passed the isnr and energy of % the objective function is displayed inside the inner loop. % Default: 'no' % info_energyo ({'yes','no'}) if 'yes' is passed the energy of % the objective function is displayed at each outter loop. % Default: 'no' % info_ISNRi ({'yes','no'}) if 'yes' is passed the isnr is displayed. % Default: 'no' % Requires: image % info_ISNRo ({'yes','no'}) if 'yes' is passed the isnr is displayed. % Default: 'no' % Requires: image % x_0 (double) initial image iteration. % Default: Wiener filter % method ({'cg','soim'}) Selects the method used to solve the linear % system: cg=conjugate gradient; soim=second order iterative % method. % Default: 'cg' % l_min (double) Minimum eigenvalue of the system for the second % order iterative method. % Default: 1e-5 % l_max (double) Maximum eigenvalue of the system for the second % order iterative method. % Default: 1 % % =================================================================== % The following functions can be provided in order to overwrite the % internal ones. Recall that, by default, all calculations are % performed with circular convolutions. (see Technical Report) % =================================================================== % % mult_H (function_handle) Function handle to the function that % performes H*x. This function must accept two parameters: % x : image to apply the convolution kernel h % h : Blur kernel % mult_Ht (function_handle) Function handle to the function that % performes Ht*x (Ht = H transpose). This function must % accept two parameters: % x : image to apply the convolution kernel h % h : Blur kernel % diffh (function_handle) Function handle to the function that % computes the horizontal differences of an image x. % diffv (function_handle) Function handle to the function that % computes the vertical differences of an image x. % diffht (function_handle) Function handle to the function that % computes the horizontal differences (transposed) of an % image x. % diffh (function_handle) Function handle to the function that % computes the vertical differences (transposed) of an % image x. % % % ====================== Output parameters =============================== % img_estimated Estimated image % energyi Energy of inner loop iterations. % Required input arguments: 'info_energyi' and 'image'. % ISNRi Improvement Signal-to-noise ratio of inner loop iterations. % Required input arguments: 'info_ISNRi' and 'image'. % energyo Energy of outer loop iterations. % Required input arguments: 'info_energyo' and 'image'. % ISNRo Improvement Signal-to-noise ratio of outer loop iterations. % Required input arguments: 'info_ISNRo' and 'image'. % % ============= EXAMPLES ========================================= % Normal use: % [img_estimated]=tvmm_debluring(img_noisy,lambda,sigma) % % Display restored images along CG iterations: % [img_estimated]=tvmm_debluring(img_noisy,lambda,sigma,... % 'displayIm','yes') % % Perform ISNR calculations of the outer loop iterations: % [img_estimated,energyi,ISNRi,energyo,ISNRo]=tvmm_debluring(img_noisy,... % lambda,sigma,'info_SNRo','yes','image',image); % % ======================= DEFAULT PARAMETERS =================== global mH mHt diffh diffv diffht diffvt boa_iter = 20; % Number of outer loop iterations cg_iter = 200; % Number of inner CG iterations soim_iter = 20; % Number of inner SOIM iterations soim_cg_iter = 6; % Number of CG iterations inside SOIM cg_thrld = 1e-5; % CG threshold displayIm = 0; image=[]; info_energyi='no'; info_energyo='no'; info_ISNRi='no'; info_ISNRo='no'; info_int = 0; info_ext = 0; energyi=0; ISNRi=0; energyo=0; ISNRo=0; mH=@conv2c; mHt=@conv2c; diffh=@f_diffh; diffv=@f_diffv; diffht=@f_diffht; diffvt=@f_diffvt; method='cg'; x_0=[]; l_min=1e-5; l_max=1; % ====================== INPUT PARAMETERS ======================== % Test for number of required parametres if (nargin-length(varargin)) ~= 3 error('Wrong number of required parameters'); end % Read the optional parameters if (rem(length(varargin),2)==1) error('Optional parameters should always go by pairs'); else for i=1:2:(length(varargin)-1) % change the value of parameter switch varargin{i} case 'boa_iter' % Outer loop iterations boa_iter = varargin{i+1}; case 'cg_iter' % Inner loop iterations cg_iter = varargin{i+1}; case 'soim_iter' % Inner loop iterations soim_iter = varargin{i+1}; case 'soim_cg_iter' % CG iterations in Inner loop soim_cg_iter = varargin{i+1}; case 'displayIm' % display sucessive xe estimates if (isequal(varargin{i+1},'yes')) displayIm=1; end case 'cg_thrld' % CG threshold cg_thrld = varargin{i+1}; case 'image' % Original image image = varargin{i+1}; case 'info_ISNRi' % display ISNR inside inner loop info_ISNRi = varargin{i+1}; case 'info_ISNRo' % display ISNR on outer loop