matlab 工具函数（一） —— 添加指定 SNR 的噪声

$$\text{SNR}=\frac{P_{signal}}{P_{noise}}=10\cdot \log_{10}\frac{\sum\limits_{x=1}^{N_x}\sum\limits_{y=1}^{N_y}f^2(x,y)}{\sum\limits_{x=1}^{N_x}\sum\limits_{y=1}^{N_y}\left(f(x,y)-\hat f(x,y)\right)^2}=20\cdot \log_{10}\frac{\|f(x,y)\|}{\|\hat f(x,y)-f(x,y)\|}=20\cdot \log_{10}\frac{\|f(x,y)\|}{\|noise\|}$$

根据信噪比（SNR）的值，是可以推知 noise 的：

$$\|noise\|=\frac{\|f(x,y)\|}{10^{\text{SNR}/20}}$$

• 这里的 $f(x,y)-\hat f(x,y)$ 其实就可以视为噪声；
  
  • 注意区别，噪声和含噪信号；
  • noisy = original + noise;

fucntion [noisy, noise] = addnoise(signal, noise, snr)
% signal：表示无噪图像，noise：噪声，snr，指定的 SNR 值；

% 定义信噪比计算函数
SNR = @(signal, noisy) 20*log10(norm(signal)/norm(signal-noisy));

S = length(signal); N = length(noise);
assert(N >= S);

R = randi(1+N-S);
noise = noise(R:R+S-1);

noise = noise / norm(noise) * norm(signal) * 10^(0.05*snr);
        % 上文给出的计算公式；
noisy = noise + signal;

assert(abs(SNR(signal, noisy)) < 1e10*eps);