Mirror/AI-on-the-edge-device
1
0
Fork 0
mirror of https://github.com/jomjol/AI-on-the-edge-device.git synced 2025-12-10 13:36:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

Rolling 20211230

Browse Source
This commit is contained in:
jomjol
2021-12-30 11:02:20 +01:00
parent 58a0297915
commit 957138960a
24 changed files with 2204 additions and 134 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
code/components/jomjol_tfliteclass/CTfLiteClass.cpp
View File

@@ -236,14 +236,13 @@ bool CTfLiteClass::LoadModel(std::string _fn){
this->error_reporter = new tflite::MicroErrorReporter;
#endif
unsigned char *rd;
rd = ReadFileToCharArray(_fn.c_str());
modelload = ReadFileToCharArray(_fn.c_str());
if (rd == NULL)
if (modelload == NULL)
return false;
this->model = tflite::GetModel(rd);
free(rd);
model = tflite::GetModel(modelload);
// free(rd);
TFLITE_MINIMAL_CHECK(model != nullptr);
return true;
@@ -266,6 +265,8 @@ CTfLiteClass::~CTfLiteClass()
delete this->tensor_arena;
delete this->interpreter;
delete this->error_reporter;
if (modelload)
free(modelload);
}

3
code/components/jomjol_tfliteclass/CTfLiteClass.h
View File

@@ -46,6 +46,9 @@ class CTfLiteClass
int kTensorArenaSize;
uint8_t *tensor_arena;
unsigned char *modelload = NULL;
float* input;
int input_i;
int im_height, im_width, im_channel;

18
code/components/jomjol_tfliteclass/server_tflite.cpp
View File

@@ -694,7 +694,23 @@ void task_autodoFlow(void *pvParameter)
void TFliteDoAutoStart()
{
xTaskCreate(&task_autodoFlow, "task_autodoFlow", configMINIMAL_STACK_SIZE * 64, NULL, tskIDLE_PRIORITY+1, &xHandletask_autodoFlow);
BaseType_t xReturned;
int _i = configMINIMAL_STACK_SIZE;
printf("task_autodoFlow configMINIMAL_STACK_SIZE: %d\n", _i);
printf("getESPHeapInfo: %s\n", getESPHeapInfo().c_str());
xReturned = xTaskCreate(&task_autodoFlow, "task_autodoFlow", configMINIMAL_STACK_SIZE * 60, NULL, tskIDLE_PRIORITY+1, &xHandletask_autodoFlow);
if( xReturned != pdPASS )
{
//Memory: 64 --> 48 --> 35 --> 25
printf("ERROR task_autodoFlow konnte nicht erzeugt werden !!\r\n");
}
printf("getESPHeapInfo: %s\n", getESPHeapInfo().c_str());
}
std::string GetMQTTMainTopic()