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

Standardize pixel values for face input

Browse Source
This commit is contained in:
Nicolas Mowen 2024-11-06 07:46:14 -07:00
parent ade1e9f0aa
commit d353450f1c
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/embeddings/functions/onnx.py
View File

@ -221,6 +221,9 @@ class GenericONNXEmbedding:
# copy img image into center of result image # copy img image into center of result image
frame[y_center : y_center + og_h, x_center : x_center + og_w] = og frame[y_center : y_center + og_h, x_center : x_center + og_w] = og
# standardize pixel values across channels
mean, std = frame.mean(), frame.std()
frame = (frame - mean) / std
frame = np.expand_dims(frame, axis=0) frame = np.expand_dims(frame, axis=0)
return [{"input_2": frame}] return [{"input_2": frame}]
elif self.model_type == ModelTypeEnum.lpr_detect: elif self.model_type == ModelTypeEnum.lpr_detect:

4
frigate/embeddings/maintainer.py
View File

@ -406,7 +406,7 @@ class EmbeddingMaintainer(threading.Thread):
logger.debug("Detected no faces for person object.") logger.debug("Detected no faces for person object.")
return return
margin = int((face_box[2] - face_box[0]) * 0.4) margin = int((face_box[2] - face_box[0]) * 0.25)
face_frame = person[ face_frame = person[
max(0, face_box[1] - margin) : min( max(0, face_box[1] - margin) : min(
frame.shape[0], face_box[3] + margin frame.shape[0], face_box[3] + margin
@ -442,7 +442,7 @@ class EmbeddingMaintainer(threading.Thread):
return return
face_frame = cv2.cvtColor(frame, cv2.COLOR_YUV2BGR_I420) face_frame = cv2.cvtColor(frame, cv2.COLOR_YUV2BGR_I420)
margin = int((face_box[2] - face_box[0]) * 0.4) margin = int((face_box[2] - face_box[0]) * 0.25)
face_frame = face_frame[ face_frame = face_frame[
max(0, face_box[1] - margin) : min( max(0, face_box[1] - margin) : min(