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);
}