File size: 1,008 Bytes
d5ee97c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
#ifndef TFLITEBASE_H
#define TFLITEBASE_H
#include "tensorflow/lite/interpreter.h"
#include "tensorflow/lite/kernels/register.h"
#include "tensorflow/lite/model.h"
#include "tensorflow/lite/optional_debug_tools.h"
#define TFLITE_MINIMAL_CHECK(x) \
if (!(x)) { \
fprintf(stderr, "Error at %s:%d\n", __FILE__, __LINE__); \
exit(1); \
}
typedef struct
{
float *melData;
std::vector<int32_t> melShape;
int32_t bytes;
} MelGenData;
class TfliteBase
{
public:
uint32_t int_size = sizeof(int32_t);
uint32_t float_size = sizeof(float);
std::unique_ptr<tflite::Interpreter> interpreter;
TfliteBase(const char* modelFilename);
~TfliteBase();
private:
std::unique_ptr<tflite::FlatBufferModel> model;
tflite::ops::builtin::BuiltinOpResolver resolver;
void interpreterBuild(const char* modelFilename);
};
#endif // TFLITEBASE_H |