Python cIRM (Complex Ideal Ratio Mask), ORM (Optimal Ratio Mask), IRM (Ideal Ratio Mask), PSM (Phase Sensitive Mask)

cirm.py

def cirm(y, s, K=10, C=0.1, flat=True):
    y = librosa.core.stft(y.astype('float64'), 256, 64).astype('complex128')
    s = librosa.core.stft(s.astype('float64'), 256, 64).astype('complex128')
    mr = (np.real(y) * np.real(s) + np.imag(y) * np.imag(s))/(np.real(y)**2 + np.imag(y)**2)
    mi = (np.real(y) * np.imag(s) - np.imag(y) * np.real(s))/(np.real(y)**2 + np.imag(y)**2)
    m = mr + 1j * mi
    if flat:
        return m
    else:
        return K * ((1 - np.exp(-C * m))/(1 + np.exp(-C * m)))


def inverse_mask(x, m, K=10, C=0.1, flat=True):
    if flat:
        return x * m
    else:
        return -1/C * np.log((K - x)/(K+x))


def psm(y, s):
    y = librosa.core.stft(y.astype('float64'), 256, 64).astype('complex128')
    s = librosa.core.stft(s.astype('float64'), 256, 64).astype('complex128')
    return np.abs(y) / np.abs(s) * np.cos(np.angle(y) + np.angle(s))


def orm(y, n, K=10, C=0.1, flat=False):
    y = librosa.core.stft(y.astype('float64'), 256, 64).astype('complex128')
    n = librosa.core.stft(n.astype('float64'), 256, 64).astype('complex128')
    m = (np.abs(y)**2 + np.real(y * np.conj(n))) / (np.abs(y)**2 + np.abs(n)**2 + 2 * np.real(y * np.conj(n)))
    if flat:
        return m
    else:
        return K * ((1 - np.exp(-C * m))/(1 + np.exp(-C * m)))


def irm(y, n):
    y = librosa.core.stft(y.astype('float64'), 256, 64).astype('complex128')
    n = librosa.core.stft(n.astype('float64'), 256, 64).astype('complex128')
    return (np.abs(y) ** 2 / (np.abs(y) ** 2 + np.abs(n) ** 2)) ** 0.5