blakeblackshear.frigate/frigate/detectors/plugins
Rémi Bédard-Couture 592b645231
Add support for TensorRT v10 (multiple api calls have changed) (#11166)
* Add support for TensorRT v10 (multiple api calls have changed)

* Remove unnecessary size check in TensorRT v10 block

* Refactor to reduce code duplication

* Fix wrong function name in new _get_binding_dtype function and only return input check (not assertion) in new _binding_is_input function

* Add space around TRT_VERSION variable assignment (=) to respect linting

* More linting fix

* Update frigate/detectors/plugins/tensorrt.py

Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com>

* More linting

---------

Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com>
2024-05-22 06:57:52 -06:00
..
__init__.py Convert detectors to factory pattern, ability to set different model for each detector (#4635) 2022-12-15 07:12:52 -06:00
cpu_tfl.py Add isort and ruff linter (#6575) 2023-05-29 05:31:17 -05:00
deepstack.py Don't fail if deepstack detector times out (#8979) 2023-12-15 23:25:21 +00:00
edgetpu_tfl.py Remove all AGPL licensed YOLO references from Frigate (#10717) 2024-03-30 05:46:17 -05:00
onnx.py Remove all AGPL licensed YOLO references from Frigate (#10717) 2024-03-30 05:46:17 -05:00
openvino.py Remove all AGPL licensed YOLO references from Frigate (#10717) 2024-03-30 05:46:17 -05:00
rknn.py Reimplement support for rknn detector (#11365) 2024-05-21 17:50:03 -05:00
rocm.py Remove all AGPL licensed YOLO references from Frigate (#10717) 2024-03-30 05:46:17 -05:00
tensorrt.py Add support for TensorRT v10 (multiple api calls have changed) (#11166) 2024-05-22 06:57:52 -06:00