avfilter/af_arnndn: make sure that vad_output nb_neurons is always 1

fb7b222b · Paul B Mahol · 1e35519f · fb7b222b
Commit fb7b222b authored Oct 17, 2019 by Paul B Mahol
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

af_arnndn.c libavfilter/af_arnndn.c +5 -0

No files found.
--- a/libavfilter/af_arnndn.c
+++ b/libavfilter/af_arnndn.c
@@ -300,6 +300,11 @@ static RNNModel *rnnoise_model_from_file(FILE *f)
    INPUT_DENSE(denoise_output);
    INPUT_DENSE(vad_output);

+    if (vad_output->nb_neurons != 1) {
+        rnnoise_model_free(ret);
+        return NULL;
+    }
+
    return ret;
 }