Skip to content

Commit c0a49ce

Browse files
committed
Output VAD prob
1 parent 98f8bd3 commit c0a49ce

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

include/rnnoise.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,4 +49,4 @@ RNNOISE_EXPORT DenoiseState *rnnoise_create();
4949

5050
RNNOISE_EXPORT void rnnoise_destroy(DenoiseState *st);
5151

52-
RNNOISE_EXPORT void rnnoise_process_frame(DenoiseState *st, float *out, const float *in);
52+
RNNOISE_EXPORT float rnnoise_process_frame(DenoiseState *st, float *out, const float *in);

src/denoise.c

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -467,7 +467,7 @@ void pitch_filter(kiss_fft_cpx *X, const kiss_fft_cpx *P, const float *Ex, const
467467
}
468468
}
469469

470-
void rnnoise_process_frame(DenoiseState *st, float *out, const float *in) {
470+
float rnnoise_process_frame(DenoiseState *st, float *out, const float *in) {
471471
int i;
472472
kiss_fft_cpx X[FREQ_SIZE];
473473
kiss_fft_cpx P[WINDOW_SIZE];
@@ -477,7 +477,7 @@ void rnnoise_process_frame(DenoiseState *st, float *out, const float *in) {
477477
float features[NB_FEATURES];
478478
float g[NB_BANDS];
479479
float gf[FREQ_SIZE]={1};
480-
float vad_prob;
480+
float vad_prob = 0;
481481
int silence;
482482
static const float a_hp[2] = {-1.99599, 0.99600};
483483
static const float b_hp[2] = {-2, 1};
@@ -497,6 +497,7 @@ void rnnoise_process_frame(DenoiseState *st, float *out, const float *in) {
497497
}
498498

499499
frame_synthesis(st, out, X);
500+
return vad_prob;
500501
}
501502

502503
#if TRAINING

0 commit comments

Comments
 (0)