fix noiseFilter

actondev · Nov 4, 2023 · fbaa24b · fbaa24b
1 parent c1d9088
commit fbaa24b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/denoise.py b/src/denoise.py
@@ -168,7 +168,10 @@ def noiseFilter(self, Px, Pn, ak):
         Hd = numpy.zeros(nSamples)
         # iterating through the 'w' (from the paper)
         for i in range(0, nSamples):
-            Hd[i] = pow((1 - self.c * Pn * ak / Px[i]), self.b)
+            if Px[i] == 0:
+                Hd[i] = 1
+            else:
+                Hd[i] = pow((1 - self.c * Pn * ak / Px[i]), self.b)
         return Hd
 
     def linearAk(self, nBands, slope):