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

"fix: correct ArcFace input format for RKNN runner"

Browse Source
This commit is contained in:
Roki 2026-03-07 20:46:47 +00:00
parent f316244495
commit 2a051a307e
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
frigate/detectors/detection_runners.py
View File

@ -529,6 +529,15 @@ class RKNNModelRunner(BaseModelRunner):
# Transpose from NCHW to NHWC
pixel_data = np.transpose(pixel_data, (0, 2, 3, 1))
rknn_inputs.append(pixel_data)
elif name == "data":
# ArcFace: undo Python normalisation to uint8 [0,255]
# RKNN runtime applies mean=127.5/std=127.5 internally before first layer
face_data = inputs[name]
if len(face_data.shape) == 4 and face_data.shape[1] == 3:
# Transpose from NCHW to NHWC
face_data = np.transpose(face_data, (0, 2, 3, 1))
face_data = ((face_data + 1.0) * 127.5).clip(0, 255).astype(np.uint8)
rknn_inputs.append(face_data)
else:
rknn_inputs.append(inputs[name])