Early discard extracted faces below threshold

This commit is contained in:
Andrea Lanfranchi 2024-02-21 16:28:07 +01:00
parent 051a8d3172
commit c4b48ce46f
No known key found for this signature in database
GPG Key ID: 2D623A60BB279915

View File

@ -91,6 +91,8 @@ def analysis(
faces = []
for face_obj in face_objs:
facial_area = face_obj["facial_area"]
if facial_area["w"] <= 130: # discard small detected faces
continue
faces.append(
(
facial_area["x"],
@ -110,8 +112,6 @@ def analysis(
detected_faces = []
face_index = 0
for x, y, w, h in faces:
if w <= 130: # discard small detected faces
continue
face_detected = True
if face_index == 0:
face_included_frames += 1 # increase frame for a single face