From 7fa6a0c8ba396c7e35f7d9c122b7ab084cbc0797 Mon Sep 17 00:00:00 2001 From: MarcA711 <> Date: Thu, 2 Nov 2023 09:19:03 +0000 Subject: [PATCH] apply changes from isort and ruff --- frigate/detectors/plugins/rknn.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) 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.")