merged nt.utils.NoiseGenerator with nt.speech_enhancement.noise chang…

…ed nt.evaluate.input_sxr to use power instead of var like in nt.speech_enhancement.noise
Haibit · Aug 16, 2019 · 6d6b412 · 6d6b412
1 parent c3c1d60
commit 6d6b412
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/nt/speech_enhancement/noise.py → nt/speech_enhancement/noise/utils.py b/nt/speech_enhancement/noise.py → nt/speech_enhancement/noise/utils.py
@@ -1,6 +1,10 @@
 import numpy
 
 
+def get_power(x, axis=None):
+    return numpy.sum(numpy.abs(x * x.conj()), axis=axis)
+
+
 def get_snr(X, N):
     """
     Return SNR of time signals or STFT signals in dB.