49a3ebf5f465bf4310cc55dc7e81ec2994c10154,paderbox/speech_enhancement/beamformer.py,,get_mvdr_vector_souden,#,627

Before Change

                "Considering an independent dim in the SNR estimate is not "
                "unique."
            )
        SNR = np.einsum(
            "...FdR,...FdD,...FDR->...R", mat.conj(), target_psd_matrix, mat
        ) / np.maximum(np.einsum(
            "...FdR,...FdD,...FDR->...R", mat.conj(), noise_psd_matrix, mat
        ), eps)
        // Raises an exception when np.inf and/or np.NaN was in target_psd_matrix
        // or noise_psd_matrix
        assert np.all(np.isfinite(SNR)), SNR
        ref_channel = np.argmax(SNR.real)

    assert np.isscalar(ref_channel), ref_channel
    beamformer = mat[..., ref_channel]

After Change

        eps = np.finfo(lambda_.dtype).tiny
    mat = phi / np.maximum(lambda_.real, eps)
    
    if ref_channel is None:
        get_optimal_reference_channel(mat, target_psd_matrix, noise_psd_matrix,
                                      eps=eps)

    assert np.isscalar(ref_channel), ref_channel
    beamformer = mat[..., ref_channel]

    if return_ref_channel:

Instances