diff --git a/matlab/tfgm/tf_fading/solver_tfgm_zero.m b/matlab/tfgm/tf_fading/solver_tfgm_zero.m deleted file mode 100644 index 9eef53133c043983b955f8b99a39721723f6bc9d..0000000000000000000000000000000000000000 --- a/matlab/tfgm/tf_fading/solver_tfgm_zero.m +++ /dev/null @@ -1,19 +0,0 @@ -function x_zero= solver_tfgm_zero(X, mask, idgt) - -%% x_zero= solver_tfgm_zero(X, mask, idgt) -% Thid function reconstruct the signal after filling the masked regions by zeros. -% Inputs: -% - X: time-frequency matrix -% - mask: binary mask -% - idgt: Inverse of Gabor transform operator -% Outputs: -% -x_interp: estimated signal -% -% -% Author: Marina KREME - -%% -X(mask==1)=0; -x_zero = compute_idgt(X, idgt); - -end \ No newline at end of file diff --git a/matlab/tfgm/tf_fading/zero_fill_solver.m b/matlab/tfgm/tf_fading/zero_fill_solver.m new file mode 100644 index 0000000000000000000000000000000000000000..c962e68dc061a64d0558956ee6d5cae93330e9ed --- /dev/null +++ b/matlab/tfgm/tf_fading/zero_fill_solver.m @@ -0,0 +1,37 @@ +function x_est= zero_fill_solver(x, mask, dgt, idgt, dgt_params,..., + signal_params, fig_dir) + +%% x_est= zero_fill_solver(x, mask, dgt, idgt, dgt_params,..., +% signal_params, fig_dir) +% Thid function reconstruct the signal after filling the masked regions by zeros. +% Inputs: +% - x (nd array): mix signals +% - mask: time-frequency mask +% - idgt,dgt (handle): DGT and IDGT . see utils/get_stft_operators.m +% - dgt_params (struct) : DGT parameters +% - signal_params (struct) . :signals parameters +% - fig_dir : folder where figures are stored +% Outputs: +% -x_est (nd array): estimated signal +% +% +% Author: Marina KREME + +%% +x_tf = dgt(x); +x_tf(mask==1)=0; +x_est = idgt(x_tf); + +%% + +figure +plot_spectrogram(x_tf, dgt_params,signal_params, dgt); +title('TF matrix filled by zeros in mask') +saveas(gcf,fullfile(fig_dir,'tf_filled_zeros.pdf')); + +figure +plot_spectrogram(x_est, dgt_params,signal_params, dgt); +title('Reconstructed signal by zero fill') +saveas(gcf,fullfile(fig_dir,'zero_fill_est.pdf')); + +end \ No newline at end of file