fix rknn.py (blakeblackshear#8434)

Co-authored-by: MarcA711 <>
gannino · Nov 3, 2023 · b54aaad · b54aaad
1 parent fc36be4
commit b54aaad
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/frigate/detectors/plugins/rknn.py b/frigate/detectors/plugins/rknn.py
@@ -6,7 +6,6 @@
 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
@@ -37,6 +36,8 @@ def __init__(self, config: RknnDetectorConfig):
 
  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.")