diff --git a/frigate/detectors/plugins/rknn.py b/frigate/detectors/plugins/rknn.py index 9a2de2daf..687e968e6 100644 --- a/frigate/detectors/plugins/rknn.py +++ b/frigate/detectors/plugins/rknn.py @@ -1,13 +1,15 @@ import logging -import numpy as np +from typing import Literal + import cv2 import cv2.dnn +import numpy as np from hide_warnings import hide_warnings +from pydantic import Field +from rknnlite.api import RKNNLite from frigate.detectors.detection_api import DetectionApi from frigate.detectors.detector_config import BaseDetectorConfig -from typing import Literal -from pydantic import Field logger = logging.getLogger(__name__) @@ -35,8 +37,6 @@ class Rknn(DetectionApi): self.model_path = config.model.path or "/models/yolov8n-320x320.rknn" - from rknnlite.api import RKNNLite - self.rknn = RKNNLite(verbose=False) if self.rknn.load_rknn(self.model_path) != 0: logger.error("Error initializing rknn model.")