• Zhaoheng Ni's avatar
    Add RTFMVDR module (#2368) · 4b021ae3
    Zhaoheng Ni authored
    Summary:
    Add a new design of MVDR module.
    The RTFMVDR module supports the method based on the relative transfer function (RTF) and power spectral density (PSD) matrix of noise.
    The input arguments are:
    - multi-channel spectrum.
    - RTF vector of the target speech
    - PSD matrix of noise.
    - reference channel in the microphone array.
    - diagonal_loading option to enable or disable diagonal loading in matrix inverse computation.
    - diag_eps for computing the inverse of the matrix.
    - eps for computing the beamforming weight.
    The output of the module is the single-channel complex-valued spectrum for the enhanced speech.
    
    Pull Request resolved: https://github.com/pytorch/audio/pull/2368
    
    Reviewed By: carolineechen
    
    Differential Revision: D36214940
    
    Pulled By: nateanl
    
    fbshipit-source-id: 5f29f778663c96591e1b520b15f7876d07116937
    4b021ae3
transforms.rst 3.03 KB