diff --git a/deepface/detectors/Yolo.py b/deepface/detectors/Yolo.py index debafc3..53e25ba 100644 --- a/deepface/detectors/Yolo.py +++ b/deepface/detectors/Yolo.py @@ -73,10 +73,17 @@ class YoloClient(Detector): # For each face, extract the bounding box, the landmarks and confidence for result in results: + + if result.boxes 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()