diff options
| author | Gregor Richards <hg-yff@gregor.im> | 2018-09-20 21:51:38 -0400 |
|---|---|---|
| committer | Jean-Marc Valin <jmvalin@jmvalin.ca> | 2019-05-29 00:37:07 -0400 |
| commit | 5e7af83876dd413e16e702269a90b4692299a720 (patch) | |
| tree | 1aede657057e4c51fbea80c04b8139553efc3480 /include | |
| parent | f30741bed8495e164049a495de89ac417f27ccf0 (diff) | |
| download | rnnoise-5e7af83876dd413e16e702269a90b4692299a720.tar.gz | |
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 *.
Diffstat (limited to 'include')
| -rw-r--r-- | include/rnnoise.h | 8 |
1 files changed, 7 insertions, 1 deletions
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 <stdio.h> + + #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 |