From 5e7af83876dd413e16e702269a90b4692299a720 Mon Sep 17 00:00:00 2001 From: Gregor Richards Date: Thu, 20 Sep 2018 21:51:38 -0400 Subject: Neural network model files Extending the neural network dumper to dump to a simple text file format, and adding reader functions to read a neural network description from a FILE *. --- include/rnnoise.h | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'include') diff --git a/include/rnnoise.h b/include/rnnoise.h index f17f8e9..67f0b06 100644 --- a/include/rnnoise.h +++ b/include/rnnoise.h @@ -28,6 +28,9 @@ #ifndef RNNOISE_H #define RNNOISE_H 1 +#include + + #ifndef RNNOISE_EXPORT # if defined(WIN32) # if defined(RNNOISE_BUILD) && defined(DLL_EXPORT) @@ -42,7 +45,6 @@ # endif #endif - typedef struct DenoiseState DenoiseState; typedef struct RNNModel RNNModel; @@ -56,4 +58,8 @@ RNNOISE_EXPORT void rnnoise_destroy(DenoiseState *st); RNNOISE_EXPORT float rnnoise_process_frame(DenoiseState *st, float *out, const float *in); +RNNOISE_EXPORT RNNModel *rnnoise_model_from_file(FILE *f); + +RNNOISE_EXPORT void rnnoise_model_free(RNNModel *model); + #endif -- cgit v1.2.3