Energy was wrongly calculated in get_snr.

Haibit · Aug 16, 2019 · fe5829b · fe5829b
1 parent f705ed4
commit fe5829b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nt/speech_enhancement/noise/utils.py b/nt/speech_enhancement/noise/utils.py
@@ -17,8 +17,8 @@ def get_snr(X, N):
     :param N: Signal of noise image.
     :return: SNR of time signals or STFT signals in dB.
     """
-    energy_X = numpy.mean(numpy.abs(X)**2)
-    energy_N = numpy.mean(numpy.abs(N)**2)
+    energy_X = numpy.var(X)
+    energy_N = numpy.var(N)
     return 10 * numpy.log10(energy_X / energy_N)