github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 20:15:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Clean the new wrapper.

Browse Source
This commit is contained in:
Vincent STRAGIER 2023-05-23 20:53:29 +02:00
parent 1b33865e05
commit b08695ac39
1 changed files with 4 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
deepface/detectors/Yolov8nfaceWrapper.py
View File

@ -28,30 +28,18 @@ def detect_face(face_detector, img, align=False):
for result in results: for result in results:
x, y, w, h = result.boxes.xywh.tolist()[0] x, y, w, h = result.boxes.xywh.tolist()[0]
confidence = result.boxes.conf.tolist()[0] confidence = result.boxes.conf.tolist()[0]
# print(f"Confidence: {confidence}, x: {x}, y: {y}, w: {w}, h: {h}")
# print landmarks
# print(result.keypoints.tolist())
# print(f"Left eye: {left_eye}, right eye: {right_eye}")
# add eyes landmarks to img
# import cv2
# img = cv2.circle(img, (int(left_eye[0]), int(left_eye[1])), 2, (0, 0, 255), 2)
# img = cv2.circle(img, (int(right_eye[0]), int(right_eye[1])), 2, (0, 255, 0), 2)
# change to top left corner, width, height
x, y, w, h = int(x - w / 2), int(y - h / 2), int(w), int(h) x, y, w, h = int(x - w / 2), int(y - h / 2), int(w), int(h)
detected_face = img[y : y + h, x : x + w].copy() detected_face = img[y : y + h, x : x + w].copy()
if align: if align:
# Extract landmarks
left_eye, right_eye, _, _, _ = result.keypoints.tolist() left_eye, right_eye, _, _, _ = result.keypoints.tolist()
# Check the landmarks confidence before alignment # Check the landmarks confidence before alignment
# print(f"Left eye: {left_eye[2]}, right eye: {right_eye[2]}")
if left_eye[2] > 0.5 and right_eye[2] > 0.5: if left_eye[2] > 0.5 and right_eye[2] > 0.5:
# print("Aligning face") detected_face = FaceDetector.alignment_procedure(
# print(left_eye[:2], right_eye[:2]) detected_face, left_eye[:2], right_eye[:2]
detected_face = FaceDetector.alignment_procedure(detected_face, left_eye[:2], right_eye[:2]) )
resp.append((detected_face, [x, y, w, h], confidence)) resp.append((detected_face, [x, y, w, h], confidence))