administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-05 18:55:23 +03:00
Code Issues Actions 2 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 cb0bd4f399
commit afec23ee60
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 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"}" echo "Generating the following TRT Models: ${YOLO_MODELS:="yolov4-tiny-288,yolov4-tiny-416,yolov7-tiny-416"}"
# Create output folder # Create output folder

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

@ -78,7 +78,7 @@ class TensorRtDetector(DetectionApi):
try: try:
trt.init_libnvinfer_plugins(self.trt_logger, "") 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: except OSError as e:
logger.error( logger.error(
"ERROR: failed to load libraries. %s", "ERROR: failed to load libraries. %s",