administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-07 11:45:24 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Update edgetpu_tfl.py

Browse Source
This commit is contained in:
alec-groff 2024-01-26 14:32:16 -06:00 committed by GitHub
parent df7cdecfdf
commit b1abdc2f8b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/detectors/plugins/edgetpu_tfl.py
View File

@ -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)