Mohammed Innat commited on
Commit
9c5e14a
1 Parent(s): 8eb6223

Update facedetect.py

Browse files
Files changed (1) hide show
  1. facedetect.py +3 -4
facedetect.py CHANGED
@@ -1,20 +1,19 @@
1
  import mediapipe as mp
2
  from utils import read_n_resize
3
 
4
- def mp_face_detect_fn(image):
5
  mp_face_detection = mp.solutions.face_detection
6
  mp_drawing = mp.solutions.drawing_utils
7
 
8
  with mp_face_detection.FaceDetection(
9
- min_detection_confidence=0.5, model_selection=0
10
- ) as face_detection:
11
 
12
  resized_image_array = read_n_resize(image, read=False)
13
 
14
  # Convert the BGR image to RGB and process it with MediaPipe Face Detection.
15
  results = face_detection.process(resized_image_array)
16
 
17
-
18
  annotated_image = resized_image_array.copy()
19
  for detection in results.detections:
20
  mp_drawing.draw_detection(annotated_image, detection)
 
1
  import mediapipe as mp
2
  from utils import read_n_resize
3
 
4
+ def mp_face_detect_fn(image, min_detect_conf=0.5):
5
  mp_face_detection = mp.solutions.face_detection
6
  mp_drawing = mp.solutions.drawing_utils
7
 
8
  with mp_face_detection.FaceDetection(
9
+ min_detection_confidence=min_detect_conf, model_selection=0
10
+ ) as face_detection:
11
 
12
  resized_image_array = read_n_resize(image, read=False)
13
 
14
  # Convert the BGR image to RGB and process it with MediaPipe Face Detection.
15
  results = face_detection.process(resized_image_array)
16
 
 
17
  annotated_image = resized_image_array.copy()
18
  for detection in results.detections:
19
  mp_drawing.draw_detection(annotated_image, detection)