File size: 469 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 |
#include "TfliteBase.h"
TfliteBase::TfliteBase(const char* modelFilename)
{
interpreterBuild(modelFilename);
}
TfliteBase::~TfliteBase()
{
;
}
void TfliteBase::interpreterBuild(const char* modelFilename)
{
model = tflite::FlatBufferModel::BuildFromFile(modelFilename);
TFLITE_MINIMAL_CHECK(model != nullptr);
tflite::InterpreterBuilder builder(*model, resolver);
builder(&interpreter);
TFLITE_MINIMAL_CHECK(interpreter != nullptr);
}
|