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-07 12:06:58 +03:00
Code Issues Packages Projects Releases Wiki Activity

add Digital100 model processing with output 0.0-9.9 as categorical network

Browse Source
This commit is contained in:
Frank Haverland
2022-05-30 23:22:17 +02:00
parent 00028010ee
commit bf8833eae6
2 changed files with 26 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
code/components/jomjol_flowcontroll/ClassFlowCNNGeneral.cpp
View File

@@ -109,7 +109,7 @@ string ClassFlowCNNGeneral::getReadout(int _analog = 0, bool _extendedResolution
return result;
}
if ((CNNType == DigitalHyprid))
if ((CNNType == DigitalHyprid || CNNType == Digital100))
{
int zif_akt = -1;
@@ -523,6 +523,10 @@ bool ClassFlowCNNGeneral::getNetworkParameter()
CNNType = DigitalHyprid;
printf("TFlite-Type set to DigitalHyprid\n");
break;
case 100:
CNNType = Digital100;
printf("TFlite-Type set to Digital\n");
break;
default:
printf("ERROR ERROR ERROR - tflite passt nicht zur Firmware - ERROR ERROR ERROR\n");
}
@@ -742,6 +746,26 @@ bool ClassFlowCNNGeneral::doNeuralNetwork(string time)
}
}
break;
case Digital100:
{
GENERAL[_ana]->ROI[i]->result_float = (float)tflite->GetClassFromImageBasis(GENERAL[_ana]->ROI[i]->image) /10.0;
printf("Result General(Digit)%i: %f\n", i, GENERAL[_ana]->ROI[i]->result_float);
if (isLogImage)
{
string _imagename = GENERAL[_ana]->name + "_" + GENERAL[_ana]->ROI[i]->name;
if (isLogImageSelect)
{
if (LogImageSelect.find(GENERAL[_ana]->ROI[i]->name) != std::string::npos)
LogImage(logPath, _imagename, &GENERAL[_ana]->ROI[i]->result_float, NULL, time, GENERAL[_ana]->ROI[i]->image_org);
}
else
{
LogImage(logPath, _imagename, &GENERAL[_ana]->ROI[i]->result_float, NULL, time, GENERAL[_ana]->ROI[i]->image_org);
}
}
} break;
default:
break;

1
code/components/jomjol_flowcontroll/ClassFlowCNNGeneral.h
View File

@@ -12,6 +12,7 @@ enum t_CNNType {
DigitalHyprid,
DigitalHyprid10,
DoubleHyprid10,
Digital100,
None
};