Commit 084a854c authored by Jan Kiene's avatar Jan Kiene
Browse files

fix for all-zero in both signals

parent acde5b1a
Loading
Loading
Loading
Loading
+5 −0
Original line number Diff line number Diff line
@@ -561,6 +561,8 @@ def ssnr(
    if np.all(diff_sig_norm == 0):
        return np.asarray([np.inf] * ref_sig_norm.shape[1])

    channels_identical_idx = np.sum(np.abs(diff_sig_norm), axis=0) == 0

    denom_add = 10**-13 * len_seg
    segment_counter = np.zeros(ref_sig.shape[1])
    for ref_seg, diff_seg in zip(
@@ -592,4 +594,7 @@ def ssnr(
        )
    ssnr[segment_counter == 0] = -np.inf

    # this prevents all-zero channels in both signals to be reported as -inf
    ssnr[channels_identical_idx] = np.inf

    return ssnr