administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-10-16 19:52:49 +03:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Add ability to configure model input dtype (#14659)

Browse Source 
* Add input type for dtype

* Add ability to manually enable TRT execution provider

* Formatting
This commit is contained in:
Nicolas Mowen
2024-10-29 10:28:05 -05:00
committed by GitHub
parent abd22d2566
commit 4e25bebdd0
4 changed files with 44 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frigate/detectors/plugins/onnx.py
+1 -1
View File
@@ -54,7 +54,7 @@ class ONNXDetector(DetectionApi):
logger.info(f"ONNX: {path} loaded")
def detect_raw(self, tensor_input):
def detect_raw(self, tensor_input: np.ndarray):
model_input_name = self.model.get_inputs()[0].name
tensor_output = self.model.run(None, {model_input_name: tensor_input})