administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-08-02 17:12:16 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

use new pycoral libraries

Browse Source
This commit is contained in:
Blake Blackshear
2021-08-07 15:34:55 -05:00
parent 4818c08fe2
commit dc759a3e56
3 changed files with 17 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frigate/edgetpu.py
+12 -16
View File
@@ -9,6 +9,7 @@ from abc import ABC, abstractmethod
from typing import Dict
import numpy as np
from pycoral.adapters import detect
import tflite_runtime.interpreter as tflite
from setproctitle import setproctitle
from tflite_runtime.interpreter import load_delegate
@@ -97,25 +98,20 @@ class LocalObjectDetector(ObjectDetector):
def detect_raw(self, tensor_input):
self.interpreter.set_tensor(self.tensor_input_details[0]["index"], tensor_input)
self.interpreter.invoke()
boxes = np.squeeze(
self.interpreter.get_tensor(self.tensor_output_details[0]["index"])
)
label_codes = np.squeeze(
self.interpreter.get_tensor(self.tensor_output_details[1]["index"])
)
scores = np.squeeze(
self.interpreter.get_tensor(self.tensor_output_details[2]["index"])
)
objects = detect.get_objects(self.interpreter, 0.4)
detections = np.zeros((20, 6), np.float32)
for i, score in enumerate(scores):
for i, obj in enumerate(objects):
if i == 20:
break
detections[i] = [
label_codes[i],
score,
boxes[i][0],
boxes[i][1],
boxes[i][2],
boxes[i][3],
obj.id,
obj.score,
obj.bbox.ymin,
obj.bbox.xmin,
obj.bbox.ymax,
obj.bbox.xmax,
]
return detections
frigate/video.py
+4 -4
View File
@@ -404,10 +404,10 @@ def detect(
for d in region_detections:
box = d[2]
size = region[2] - region[0]
x_min = int((box[1] * size) + region[0])
y_min = int((box[0] * size) + region[1])
x_max = int((box[3] * size) + region[0])
y_max = int((box[2] * size) + region[1])
x_min = int((box[1]) + region[0])
y_min = int((box[0]) + region[1])
x_max = int((box[3]) + region[0])
y_max = int((box[2]) + region[1])
det = (
d[0],
d[1],