Wavelet

Development Platform:

Matlab

noise.m：Code Content

							function nX = noise(varargin)
% 图像加入噪声
% 传入参数依次为:
% X - 待处理的图像
% type - 噪声类型
% variance/density - 高斯噪声的方差/椒盐噪声的密度(optional)
% M - 控制噪声区域的模板(optional)
% 参数默认值
variance = 0.01;                %高斯噪声的方差 默认值为0.01
density = 0.05;                 %椒盐噪声的密度 默认值为0.05
M = ones(size(varargin{1}));    %控制噪声区域的模板 默认对整个图像加噪声
X = varargin{1};                %待处理的图像
type = varargin{2};             %噪声类型
% 可选参数
if nargin>=3
    variance = varargin{3};
    density = varargin{3};
    
    if nargin == 4
        M = varargin{4};
    end
end
switch type
    case 'gaussian'             %高斯噪声
        n = randn(size(X)) * sqrt(variance);
        n = n .* M;
        nX = X + n;
    case 'salt & pepper'        %椒盐噪声
        nX = X;
        n = rand(size(X));
        n = n .* M;
        ind = find( n>0 & n<density/2);
        nX(ind) = 0;
        ind = find( n>=density/2 & n<density);
        nX(ind) = 1;
end

						