diff options
| author | Jean-Marc Valin <jmvalin@jmvalin.ca> | 2017-08-18 18:16:38 +0300 |
|---|---|---|
| committer | Jean-Marc Valin <jmvalin@jmvalin.ca> | 2017-08-18 18:16:38 +0300 |
| commit | 3daf3571f56c99c74cf4264747575913d5f1ea31 (patch) | |
| tree | d49eaf405940a6b67d77c9b893d2fffabdd68bd8 | |
| parent | 960b8c207c18e37fe5b95b37fdb5df7482b99805 (diff) | |
features #8
| -rw-r--r-- | src/denoise.c | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/src/denoise.c b/src/denoise.c index b37d9f8..6a42274 100644 --- a/src/denoise.c +++ b/src/denoise.c @@ -301,6 +301,7 @@ void rnnoise_destroy(DenoiseState *st) { #if TRAINING int lowpass = FREQ_SIZE; +int band_lp = NB_BANDS; #endif static void frame_analysis(DenoiseState *st, kiss_fft_cpx *X, float *Ex, const float *in) { @@ -570,6 +571,12 @@ int main(int argc, char **argv) { rand_resp(a_noise, b_noise); rand_resp(a_sig, b_sig); lowpass = FREQ_SIZE * 3000./24000. * pow(10., rand()/(double)RAND_MAX); + for (i=0;i<NB_BANDS;i++) { + if (eband5ms[i]<<FRAME_SIZE_SHIFT > lowpass) { + band_lp = i; + break; + } + } } fread(tmp, sizeof(short), FRAME_SIZE, f1); if (feof(f1)) break; @@ -605,7 +612,7 @@ int main(int argc, char **argv) { for (i=0;i<NB_BANDS;i++) { g[i] = sqrt((Ey[i]+1e-2)/(Ex[i]+1e-2)); if (g[i] > 1) g[i] = 1; - if (silence) g[i] = -1; + if (silence || i > band_lp) g[i] = -1; } count++; #if 0 |