administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-18 09:04:28 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Do not add margin to face images

Browse Source
This commit is contained in:
Nicolas Mowen 2025-01-05 08:25:12 -07:00
parent 3e587f93de
commit e7aa6c6bc2
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
frigate/embeddings/maintainer.py
View File

@ -492,15 +492,10 @@ class EmbeddingMaintainer(threading.Thread):
return False return False
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.25)
face_frame = face_frame[ face_frame = face_frame[
max(0, face_box[1] - margin) : min( max(0, face_box[1]) : min(frame.shape[0], face_box[3]),
frame.shape[0], face_box[3] + margin max(0, face_box[0]) : min(frame.shape[1], face_box[2]),
),
max(0, face_box[0] - margin) : min(
frame.shape[1], face_box[2] + margin
),
] ]
res = self.face_classifier.classify_face(face_frame) res = self.face_classifier.classify_face(face_frame)