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

Fix logical errors caused by code formatting

Browse Source
This commit is contained in:
shizhicheng 2025-11-11 05:54:19 +00:00
parent 1dee548dbc
commit e27a94ae0b
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -21,7 +21,7 @@ logger = logging.getLogger(__name__)
DETECTOR_KEY = "axengine" DETECTOR_KEY = "axengine"
supported_models = { supported_models = {
ModelTypeEnum.yologeneric: "frigate-yolov9-tiny", ModelTypeEnum.yologeneric: "frigate-yolov9-.*$",
} }
model_cache_dir = os.path.join(MODEL_CACHE_DIR, "axengine_cache/") model_cache_dir = os.path.join(MODEL_CACHE_DIR, "axengine_cache/")
@ -38,9 +38,7 @@ class Axengine(DetectionApi):
self.height = config.model.height self.height = config.model.height
self.width = config.model.width self.width = config.model.width
model_path = config.model.path or "frigate-yolov9-tiny" model_path = config.model.path or "frigate-yolov9-tiny"
model_props = self.parse_model_input(model_path) model_props = self.parse_model_input(model_path)
self.session = axe.InferenceSession(model_props["path"]) self.session = axe.InferenceSession(model_props["path"])
def __del__(self): def __del__(self):
@ -51,6 +49,7 @@ class Axengine(DetectionApi):
model_props["preset"] = True model_props["preset"] = True
model_matched = False model_matched = False
for model_type, pattern in supported_models.items(): for model_type, pattern in supported_models.items():
if re.match(pattern, model_path): if re.match(pattern, model_path):
model_matched = True model_matched = True