diff options
author | Jean-Marc Valin <jmvalin@jmvalin.ca> | 2017-09-13 19:56:46 -0400 |
---|---|---|
committer | Jean-Marc Valin <jmvalin@jmvalin.ca> | 2017-09-13 19:56:46 -0400 |
commit | c0a49cea32898c4805a7a94c96905f494a2aa03f (patch) | |
tree | bbe03cf349d8d8adf8d28c1e3a1dd7465b9c7262 | |
parent | 98f8bd3158507de10c7788edb983013ea6835d01 (diff) | |
download | rnnoise-c0a49cea32898c4805a7a94c96905f494a2aa03f.tar.gz |
Output VAD prob
-rw-r--r-- | include/rnnoise.h | 2 | ||||
-rw-r--r-- | src/denoise.c | 5 |
2 files changed, 4 insertions, 3 deletions
diff --git a/include/rnnoise.h b/include/rnnoise.h index f85a522..562b34c 100644 --- a/include/rnnoise.h +++ b/include/rnnoise.h @@ -49,4 +49,4 @@ RNNOISE_EXPORT DenoiseState *rnnoise_create(); RNNOISE_EXPORT void rnnoise_destroy(DenoiseState *st); -RNNOISE_EXPORT void rnnoise_process_frame(DenoiseState *st, float *out, const float *in); +RNNOISE_EXPORT float rnnoise_process_frame(DenoiseState *st, float *out, const float *in); diff --git a/src/denoise.c b/src/denoise.c index 075edc8..cb545b6 100644 --- a/src/denoise.c +++ b/src/denoise.c @@ -467,7 +467,7 @@ void pitch_filter(kiss_fft_cpx *X, const kiss_fft_cpx *P, const float *Ex, const } } -void rnnoise_process_frame(DenoiseState *st, float *out, const float *in) { +float rnnoise_process_frame(DenoiseState *st, float *out, const float *in) { int i; kiss_fft_cpx X[FREQ_SIZE]; kiss_fft_cpx P[WINDOW_SIZE]; @@ -477,7 +477,7 @@ void rnnoise_process_frame(DenoiseState *st, float *out, const float *in) { float features[NB_FEATURES]; float g[NB_BANDS]; float gf[FREQ_SIZE]={1}; - float vad_prob; + float vad_prob = 0; int silence; static const float a_hp[2] = {-1.99599, 0.99600}; static const float b_hp[2] = {-2, 1}; @@ -497,6 +497,7 @@ void rnnoise_process_frame(DenoiseState *st, float *out, const float *in) { } frame_synthesis(st, out, X); + return vad_prob; } #if TRAINING |