mirror of
https://github.com/jomjol/AI-on-the-edge-device.git
synced 2025-12-06 19:46:54 +03:00
57 lines
1.4 KiB
C++
57 lines
1.4 KiB
C++
#pragma once
|
|
|
|
#ifndef CTFLITECLASS_H
|
|
#define CTFLITECLASS_H
|
|
|
|
#include "tensorflow/lite/micro/micro_mutable_op_resolver.h"
|
|
#include "tensorflow/lite/micro/micro_interpreter.h"
|
|
#include "tensorflow/lite/schema/schema_generated.h"
|
|
|
|
#include "esp_err.h"
|
|
#include "esp_log.h"
|
|
|
|
#include "CImageBasis.h"
|
|
|
|
|
|
class CTfLiteClass
|
|
{
|
|
protected:
|
|
tflite::MicroMutableOpResolver<10> resolver;
|
|
const tflite::Model* model;
|
|
tflite::MicroInterpreter* interpreter;
|
|
TfLiteTensor* output = nullptr;
|
|
|
|
int kTensorArenaSize;
|
|
uint8_t *tensor_arena;
|
|
|
|
unsigned char *modelfile = NULL;
|
|
|
|
|
|
float* input;
|
|
int input_i;
|
|
int im_height, im_width, im_channel;
|
|
|
|
long GetFileSize(std::string filename);
|
|
bool ReadFileToModel(std::string _fn);
|
|
void MakeStaticResolver();
|
|
|
|
public:
|
|
CTfLiteClass();
|
|
~CTfLiteClass();
|
|
bool LoadModel(std::string _fn);
|
|
bool MakeAllocate();
|
|
void GetInputTensorSize();
|
|
bool LoadInputImageBasis(CImageBasis *rs);
|
|
void Invoke();
|
|
int GetAnzOutPut(bool silent = true);
|
|
int GetOutClassification(int _von = -1, int _bis = -1);
|
|
|
|
int GetClassFromImageBasis(CImageBasis *rs);
|
|
std::string GetStatusFlow();
|
|
|
|
float GetOutputValue(int nr);
|
|
void GetInputDimension(bool silent);
|
|
int ReadInputDimenstion(int _dim);
|
|
};
|
|
|
|
#endif //CTFLITECLASS_H
|