48 lines
2.0 KiB
C
48 lines
2.0 KiB
C
|
#ifndef DL_LIB_COEFGETTER_IF_H
|
||
|
|
#define DL_LIB_COEFGETTER_IF_H
|
||
|
|
|
||
|
|
#include "dl_lib_matrix.h"
|
||
|
|
#include "dl_lib_matrixq.h"
|
||
|
|
#include "dl_lib_matrix3d.h"
|
||
|
|
#include "dl_lib_matrix3dq.h"
|
||
|
|
|
||
|
|
//Set this if the coefficient requested is a batch-normalization popvar matrix which needs to be preprocessed by
|
||
|
|
//dl_batch_normalize_get_sqrtvar first.
|
||
|
|
#define COEF_GETTER_HINT_BNVAR (1<<0)
|
||
|
|
|
||
|
|
/*
|
||
|
|
This struct describes the basic information of model data:
|
||
|
|
word_num: the number of wake words or speech commands
|
||
|
|
word_list: the name list of wake words or speech commands
|
||
|
|
thres_list: the threshold list of wake words or speech commands
|
||
|
|
info_str: the string used to reflect the version and information of model data
|
||
|
|
which consist of the architecture of network, the version of model data, wake words and their threshold
|
||
|
|
*/
|
||
|
|
typedef struct {
|
||
|
|
int word_num;
|
||
|
|
char **word_list;
|
||
|
|
int *win_list;
|
||
|
|
float *thresh_list;
|
||
|
|
char *info_str;
|
||
|
|
} model_info_t;
|
||
|
|
|
||
|
|
/*
|
||
|
|
This struct describes a generic coefficient getter: a way to get the constant coefficients needed for a neural network.
|
||
|
|
For the two getters, the name describes the name of the coefficient matrix, usually the same as the Numpy filename the
|
||
|
|
coefficient was originally stored in. The arg argument can be used to optionally pass an additional user-defined argument
|
||
|
|
to the getter (e.g. the directory to look for files in the case of the Numpy file loader getter). The hint argument
|
||
|
|
is a bitwise OR of the COEF_GETTER_HINT_* flags or 0 when none is needed. Use the free_f/free_q functions to release the
|
||
|
|
memory for the returned matrices, when applicable.
|
||
|
|
*/
|
||
|
|
typedef struct {
|
||
|
|
const dl_matrix2d_t* (*getter_f)(const char *name, void *arg, int hint);
|
||
|
|
const dl_matrix2dq_t* (*getter_q)(const char *name, void *arg, int hint);
|
||
|
|
const dl_matrix3d_t* (*getter_3d)(const char *name, void *arg, int hint);
|
||
|
|
const dl_matrix3dq_t* (*getter_3dq)(const char *name, void *arg, int hint);
|
||
|
|
void (*free_f)(const dl_matrix2d_t *m);
|
||
|
|
void (*free_q)(const dl_matrix2dq_t *m);
|
||
|
|
const model_info_t* (*getter_info)(void *arg);
|
||
|
|
} model_coeff_getter_t;
|
||
|
|
|
||
|
|
#endif
|