administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-07 19:55:26 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

apply changes from isort and ruff

Browse Source
This commit is contained in:
MarcA711 2023-11-02 09:19:03 +00:00
parent e3342cd2dd
commit 7fa6a0c8ba
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,13 +1,15 @@
import logging import logging
import numpy as np from typing import Literal
import cv2 import cv2
import cv2.dnn import cv2.dnn
import numpy as np
from hide_warnings import hide_warnings 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.detection_api import DetectionApi
from frigate.detectors.detector_config import BaseDetectorConfig from frigate.detectors.detector_config import BaseDetectorConfig
from typing import Literal
from pydantic import Field
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -35,8 +37,6 @@ 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.")