Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues Projects Releases Wiki Activity

fix rknn.py (#8434)

Browse Source 
Co-authored-by: MarcA711 <>
This commit is contained in:
Marc Altmann 2023-11-03 01:12:54 +01:00 committed by GitHub
parent fc36be4f88
commit b54aaad382
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/detectors/plugins/rknn.py
View File

@ -6,7 +6,6 @@ import cv2.dnn
import numpy as np import numpy as np
from hide_warnings import hide_warnings from hide_warnings import hide_warnings
from pydantic import Field from pydantic import Field
from rknnlite.api import RKNNLite
from frigate.detectors.detection_api import DetectionApi from frigate.detectors.detection_api import DetectionApi
from frigate.detectors.detector_config import BaseDetectorConfig from frigate.detectors.detector_config import BaseDetectorConfig
@ -37,6 +36,8 @@ class Rknn(DetectionApi):
self.model_path = config.model.path or "/models/yolov8n-320x320.rknn" self.model_path = config.model.path or "/models/yolov8n-320x320.rknn"
from rknnlite.api import RKNNLite
self.rknn = RKNNLite(verbose=False) self.rknn = RKNNLite(verbose=False)
if self.rknn.load_rknn(self.model_path) != 0: if self.rknn.load_rknn(self.model_path) != 0:
logger.error("Error initializing rknn model.") logger.error("Error initializing rknn model.")