From b1abdc2f8b887bf7d7eaaf4d6f8265b5d1843ce6 Mon Sep 17 00:00:00 2001 From: alec-groff <148134052+alec-groff@users.noreply.github.com> Date: Fri, 26 Jan 2024 14:32:16 -0600 Subject: [PATCH] Update edgetpu_tfl.py --- frigate/detectors/plugins/edgetpu_tfl.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/frigate/detectors/plugins/edgetpu_tfl.py b/frigate/detectors/plugins/edgetpu_tfl.py index 5b3435197..37a9ea7e4 100644 --- a/frigate/detectors/plugins/edgetpu_tfl.py +++ b/frigate/detectors/plugins/edgetpu_tfl.py @@ -60,11 +60,11 @@ class EdgeTpuTfl(DetectionApi): self.interpreter.set_tensor(self.tensor_input_details[0]["index"], tensor_input) self.interpreter.invoke() - boxes = self.interpreter.tensor(self.tensor_output_details[0]["index"])()[0] - class_ids = self.interpreter.tensor(self.tensor_output_details[1]["index"])()[0] - scores = self.interpreter.tensor(self.tensor_output_details[2]["index"])()[0] + boxes = self.interpreter.tensor(self.tensor_output_details[self.order[0]]["index"])()[0] + class_ids = self.interpreter.tensor(self.tensor_output_details[self.order[1]]["index"])()[0] + scores = self.interpreter.tensor(self.tensor_output_details[self.order[2]]["index"])()[0] count = int( - self.interpreter.tensor(self.tensor_output_details[3]["index"])()[0] + self.interpreter.tensor(self.tensor_output_details[self.order[3]]["index"])()[0] ) detections = np.zeros((20, 6), np.float32)