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

Print all available output tensors

Browse Source
This commit is contained in:
Nate Meyer 2022-10-25 22:35:56 -04:00
parent 98b280884c
commit a85468d243
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
frigate/detectors/openvino.py
View File

@ -17,7 +17,15 @@ class OvDetector(DetectionApi):
model=self.ov_model, device_name=det_device
)
logger.info(f"Model Input Shape: {self.interpreter.input(0).shape}")
logger.info(f"Model Output Shape: {self.interpreter.output(0).shape}")
self.output_indexes = 0
while True:
try:
tensor_shape = self.interpreter.output(self.output_indexes).shape
logger.info(f"Model Output-{self.output_indexes} Shape: {tensor_shape}")
self.output_indexes += 1
except:
logger.info(f"Model has {self.output_indexes} Output Tensors")
break
def detect_raw(self, tensor_input):