Mirror/AI-on-the-edge-device
1
0
Fork 0
mirror of https://github.com/jomjol/AI-on-the-edge-device.git synced 2026-01-29 05:40:40 +03:00
Code Issues Packages Projects Releases Wiki Activity

work on GPIO handling

Browse Source
This commit is contained in:
Zwer2k
2021-06-13 01:24:02 +02:00
parent c25adfe28a
commit f24c40d780
13 changed files with 330 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -208,7 +208,7 @@ unsigned char* CTfLiteClass::ReadFileToCharArray(std::string _fn)
return result;
}
void CTfLiteClass::LoadModel(std::string _fn){
bool CTfLiteClass::LoadModel(std::string _fn){
#ifdef SUPRESS_TFLITE_ERRORS
this->error_reporter = new tflite::OwnMicroErrorReporter;
@@ -219,9 +219,14 @@ void CTfLiteClass::LoadModel(std::string _fn){
unsigned char *rd;
rd = ReadFileToCharArray(_fn.c_str());
if (rd == NULL)
return false;
this->model = tflite::GetModel(rd);
free(rd);
TFLITE_MINIMAL_CHECK(model != nullptr);
return true;
}