nik/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-03-04 06:33:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change normalization for better accuracy

Browse Source
This commit is contained in:
Nicolas Mowen 2025-08-21 05:21:52 -06:00
parent ba5030a84d
commit 36b1f153e8
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
frigate/util/rknn_converter.py
View File

@ -28,12 +28,8 @@ MODEL_TYPE_CONFIGS = {
"target_platform": None, # Will be set dynamically
},
"jina-clip-v1-vision": {
"mean_values": [
[0.48145466, 0.4578275, 0.40821073]
], # CLIP standard normalization
"std_values": [
[0.26862954, 0.26130258, 0.27577711]
], # CLIP standard normalization
"mean_values": [[0.48145466 * 255, 0.4578275 * 255, 0.40821073 * 255]],
"std_values": [[0.26862954 * 255, 0.26130258 * 255, 0.27577711 * 255]],
"target_platform": None, # Will be set dynamically
},
}
@ -43,6 +39,9 @@ def get_rknn_model_type(model_path: str) -> str | None:
if all(keyword in str(model_path) for keyword in ["jina-clip-v1", "vision"]):
return "jina-clip-v1-vision"
if all(keyword in str(model_path) for keyword in ["jina-clip-v2", "vision"]):
return "jina-clip-v2-vision"
model_name = os.path.basename(str(model_path)).lower()
if any(keyword in model_name for keyword in ["yolo", "yolox", "yolonas"]):