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

simlipfy yolo result is valid

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-03-10 10:50:39 +00:00
parent c09e94ecaa
commit e13f3cf8ad
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
deepface/detectors/Yolo.py
View File

@ -74,16 +74,13 @@ class YoloClient(Detector):
# For each face, extract the bounding box, the landmarks and confidence
for result in results:
if result.boxes is None:
if result.boxes is None or result.keypoints is None:
continue
# Extract the bounding box and the confidence
x, y, w, h = result.boxes.xywh.tolist()[0]
confidence = result.boxes.conf.tolist()[0]
if result.keypoints is None:
continue
# left_eye_conf = result.keypoints.conf[0][0]
# right_eye_conf = result.keypoints.conf[0][1]
left_eye = result.keypoints.xy[0][0].tolist()