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

Fix path handling

Browse Source
This commit is contained in:
Nicolas Mowen 2025-08-20 20:39:17 -06:00
parent ba77f066a0
commit 9c8ca75ce4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/util/rknn_converter.py
View File

@ -40,10 +40,10 @@ MODEL_TYPE_CONFIGS = {
def get_rknn_model_type(model_path: str) -> str | None: def get_rknn_model_type(model_path: str) -> str | None:
if all(keyword in model_path for keyword in ["jina-clip-v1", "vision"]): if all(keyword in str(model_path) for keyword in ["jina-clip-v1", "vision"]):
return "jina-clip-v1-vision" return "jina-clip-v1-vision"
model_name = os.path.basename(model_path).lower() model_name = os.path.basename(str(model_path)).lower()
if any(keyword in model_name for keyword in ["yolo", "yolox", "yolonas"]): if any(keyword in model_name for keyword in ["yolo", "yolox", "yolonas"]):
return model_name return model_name