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

Fix paths to yolo lib and models folder

Browse Source
This commit is contained in:
Nate Meyer 2023-06-20 02:19:33 -04:00
parent 8a6b840e47
commit 7155cc4b87
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docker/tensorrt_models.sh
View File

@ -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

2
frigate/detectors/plugins/tensorrt.py
View File

@ -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",