experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

TFLu: Fix bug in PPD op

Browse Source 
Change-Id: I58da0d92491d264d6ba90fb1b3a5673c0058047f
This commit is contained in:
Måns Nilsson 2020-12-08 08:56:21 +01:00
parent 0939f0b7a8
commit 4425be2b7b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/lite/micro/kernels/detection_postprocess.cc
View File

@ -357,8 +357,9 @@ int SelectDetectionsAboveScoreThreshold(const float* values, int size,
int counter = 0; int counter = 0;
for (int i = 0; i < size; i++) { for (int i = 0; i < size; i++) {
if (values[i] >= threshold) { if (values[i] >= threshold) {
keep_values[counter++] = values[i]; keep_values[counter] = values[i];
keep_indices[i] = i; keep_indices[counter] = i;
counter++;
} }
} }
return counter; return counter;