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

ruff format

Browse Source
This commit is contained in:
Roki 2026-03-07 21:00:37 +00:00
parent 2a051a307e
commit 2fe2c81d8f
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/detectors/detection_runners.py
View File

@ -536,7 +536,9 @@ class RKNNModelRunner(BaseModelRunner):
if len(face_data.shape) == 4 and face_data.shape[1] == 3: if len(face_data.shape) == 4 and face_data.shape[1] == 3:
# Transpose from NCHW to NHWC # Transpose from NCHW to NHWC
face_data = np.transpose(face_data, (0, 2, 3, 1)) face_data = np.transpose(face_data, (0, 2, 3, 1))
face_data = ((face_data + 1.0) * 127.5).clip(0, 255).astype(np.uint8) face_data = (
((face_data + 1.0) * 127.5).clip(0, 255).astype(np.uint8)
)
rknn_inputs.append(face_data) rknn_inputs.append(face_data)
else: else:
rknn_inputs.append(inputs[name]) rknn_inputs.append(inputs[name])