administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-12-06 13:34:13 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

format code with ruff

Browse Source
This commit is contained in:
shizhicheng 2025-11-24 02:42:04 +00:00
parent b4abbd7d3b
commit f134796913
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
frigate/detectors/plugins/axengine.py
View File

@ -4,18 +4,13 @@ import re
import urllib.request import urllib.request
from typing import Literal from typing import Literal
import cv2 import axengine as axe
import numpy as np
from pydantic import Field
from frigate.const import MODEL_CACHE_DIR from frigate.const import MODEL_CACHE_DIR
from frigate.detectors.detection_api import DetectionApi from frigate.detectors.detection_api import DetectionApi
from frigate.detectors.detector_config import BaseDetectorConfig, ModelTypeEnum from frigate.detectors.detector_config import BaseDetectorConfig, ModelTypeEnum
from frigate.util.model import post_process_yolo from frigate.util.model import post_process_yolo
import axengine as axe
from axengine import axclrt_provider_name, axengine_provider_name
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
DETECTOR_KEY = "axengine" DETECTOR_KEY = "axengine"
@ -56,7 +51,7 @@ class Axengine(DetectionApi):
model_props["model_type"] = model_type model_props["model_type"] = model_type
if model_matched: if model_matched:
model_props["filename"] = model_path + f".axmodel" model_props["filename"] = model_path + ".axmodel"
model_props["path"] = model_cache_dir + model_props["filename"] model_props["path"] = model_cache_dir + model_props["filename"]
if not os.path.isfile(model_props["path"]): if not os.path.isfile(model_props["path"]):