Skip to content

Commit a42c2c6

Browse files
committed
add filtering low confidence faces in the MaskFromFaceModel node.
1 parent 1c8530b commit a42c2c6

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

py/nodes.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -622,6 +622,11 @@ def mask_get(self, image, max_face_number, add_bbox_upper_points, faceanalysis=N
622622
if len(face_models) == 0:
623623
return (cant_detect_result,)
624624

625+
# 过滤低置信度的face
626+
det_scores = [f.det_score for f in face_models]
627+
if len(face_models) > 1 and max(det_scores) > 0.7:
628+
face_models = [f for f in face_models if f.det_score > 0.7]
629+
625630
if max_face_number !=-1 and len(face_model) > max_face_number:
626631
face_models = self.remove_unavaible_face_models(face_models=face_models,max_people_number=max_face_number)
627632

0 commit comments

Comments
 (0)