From 4425be2b7b3ea85cd538f766802664cb6dfd8983 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?M=C3=A5ns=20Nilsson?= Date: Tue, 8 Dec 2020 08:56:21 +0100 Subject: [PATCH] TFLu: Fix bug in PPD op Change-Id: I58da0d92491d264d6ba90fb1b3a5673c0058047f --- tensorflow/lite/micro/kernels/detection_postprocess.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow/lite/micro/kernels/detection_postprocess.cc b/tensorflow/lite/micro/kernels/detection_postprocess.cc index cdf86349ed2..63ace6049e4 100644 --- a/tensorflow/lite/micro/kernels/detection_postprocess.cc +++ b/tensorflow/lite/micro/kernels/detection_postprocess.cc @@ -357,8 +357,9 @@ int SelectDetectionsAboveScoreThreshold(const float* values, int size, int counter = 0; for (int i = 0; i < size; i++) { if (values[i] >= threshold) { - keep_values[counter++] = values[i]; - keep_indices[i] = i; + keep_values[counter] = values[i]; + keep_indices[counter] = i; + counter++; } } return counter;