diff --git a/docker/tensorrt_models.sh b/docker/tensorrt_models.sh index a2aead7f9..27e98a078 100755 --- a/docker/tensorrt_models.sh +++ b/docker/tensorrt_models.sh @@ -2,7 +2,7 @@ set -euxo pipefail -OUTPUT_FOLDER=/trt_models +OUTPUT_FOLDER=/trt-models echo "Generating the following TRT Models: ${YOLO_MODELS:="yolov4-tiny-288,yolov4-tiny-416,yolov7-tiny-416"}" # Create output folder diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index 7251b8751..dea3fe078 100644 --- a/frigate/detectors/plugins/tensorrt.py +++ b/frigate/detectors/plugins/tensorrt.py @@ -78,7 +78,7 @@ class TensorRtDetector(DetectionApi): try: trt.init_libnvinfer_plugins(self.trt_logger, "") - ctypes.cdll.LoadLibrary("/trt-models/libyolo_layer.so") + ctypes.cdll.LoadLibrary("/usr/local/lib/libyolo_layer.so") except OSError as e: logger.error( "ERROR: failed to load libraries. %s",