3678a2dc9162cd9f6905cb717bdc938926fc1470,nussl/separation/ft2d.py,FT2D,compute_ft2d_mask,#FT2D#Any#Any#,97

Before Change

        fg_stft = np.abs(np.fft.ifft2(fg_ft2d)) / (_stft + 1e-7)
        
        bg_mask = bg_stft / (np.maximum(bg_stft, fg_stft) + 1e-7)
        fg_mask = fg_stft / (np.maximum(bg_stft, fg_stft) + 1e-7)
        fg_mask = fg_mask * (1 - bg_mask)
        bg_mask = bg_mask * (1 - fg_mask)
        return bg_mask, fg_mask

    def filter_local_maxima_with_std(self, ft2d):
        data = np.abs(np.fft.fftshift(ft2d))

After Change

            bg_ft2d, fg_ft2d = self.filter_local_maxima_with_std(ft2d[:, :, ch])
        
        _stft = np.abs(self.stft)[:, :, ch] + 1e-7
        bg_stft = np.minimum(np.abs(np.fft.ifft2(bg_ft2d)), _stft)
        
        bg_mask = bg_stft / _stft
        fg_mask = 1 - bg_mask
        return bg_mask, fg_mask

Instances