Commit Graph

6 Commits

Author SHA1 Message Date
Nate Meyer
7155cc4b87 Fix paths to yolo lib and models folder 2023-06-20 02:19:33 -04:00
Nate Meyer
8a6b840e47 Fix typo in model script 2023-06-20 02:14:09 -04:00
Nate Meyer
d244a2532c Update tensorrt_models script to convert models from the frigate container 2023-06-20 00:21:42 -04:00
Nate Meyer
6b7ef8812d Update to latest tensorrt (8.6.1) release 2023-06-17 23:57:08 -04:00
Nate Meyer
8ac3114f9a
Cleanup Detector labelmap (#4932)
* Add missing labels to default labelmap.  Fill any holes with "unknown".  Remove unique labelmap for tensorrt.

* Replace "truck" with "car" on Openvino labelmap
2023-01-06 07:03:16 -06:00
Nate Meyer
3f05f74ecb
Nvidia TensorRT detector (#4718)
* Initial WIP dockerfile and scripts to add tensorrt support

* Add tensorRT detector

* WIP attempt to install TensorRT 8.5

* Updates to detector for cuda python library

* TensorRT Cuda library rework WIP

Does not run

* Fixes from rebase to detector factory

* Fix parsing output memory pointer

* Handle TensorRT logs with the python logger

* Use non-async interface and convert input data to float32. Detection runs without error.

* Make TensorRT a separate build from the base Frigate image.

* Add script and documentation for generating TRT Models

* Add support for TensorRT devcontainer

* Add labelmap to trt model script and docs.  Cleanup of old scripts.

* Update detect to normalize input tensor using model input type

* Add config for selecting GPU. Fix Async inference. Update documentation.

* Update some CUDA libraries to clean up version warning

* Add CI stage to build TensorRT tag

* Add note in docs for image tag and model support
2022-12-30 10:53:17 -06:00