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

Support face recognition via RKNN (#19687)

Browse Source 
* Add support for face recognition via RKNN

* Fix crash when adding camera in via UI

* Update docs regarding support for face recognition

* Formatting
This commit is contained in:
Nicolas Mowen
2025-08-21 06:18:55 -06:00
committed by GitHub
parent 1be84d6833
commit f39475a383
5 changed files with 24 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frigate/util/rknn_converter.py
+14
View File
@@ -32,6 +32,11 @@ MODEL_TYPE_CONFIGS = {
"std_values": [[0.26862954 * 255, 0.26130258 * 255, 0.27577711 * 255]],
"target_platform": None, # Will be set dynamically
},
"arcface-r100": {
"mean_values": [[127.5, 127.5, 127.5]],
"std_values": [[127.5, 127.5, 127.5]],
"target_platform": None, # Will be set dynamically
},
}
@@ -41,6 +46,9 @@ def get_rknn_model_type(model_path: str) -> str | None:
model_name = os.path.basename(str(model_path)).lower()
if "arcface" in model_name:
return "arcface-r100"
if any(keyword in model_name for keyword in ["yolo", "yolox", "yolonas"]):
return model_name
@@ -184,6 +192,12 @@ def convert_onnx_to_rknn(
inputs=["pixel_values"],
input_size_list=[[1, 3, 224, 224]],
)
elif model_type == "arcface-r100":
load_output = rknn.load_onnx(
model=onnx_path,
inputs=["data"],
input_size_list=[[1, 3, 112, 112]],
)
else:
load_output = rknn.load_onnx(model=onnx_path)