03_face_detection.py

import cv2
import mediapipe as mp
import time

cap =cv2.VideoCapture(0,cv2.CAP_DSHOW)
pTime=0

mpFaceDetection=mp.solutions.face_detection
mpDraw=mp.solutions.drawing_utils
faceDetection=mpFaceDetection.FaceDetection()

while True:
    success,img=cap.read()
    imgRGB=cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
    results=faceDetection.process(imgRGB)

    if results.detections:
        for id,detection in enumerate(results.detections):
            #mpDraw.draw_detection(img,detection)
            bboxc=detection.location_data.relative_bounding_box
            ih,iw,ic=img.shape
            bbox=int(bboxc.xmin*iw),int(bboxc.ymin*ih),int(bboxc.width*iw),int(bboxc.height*ih)
            cv2.rectangle(img,bbox,(255,0,255),2)
            cv2.putText(img,f'{int(detection.score[0]*100)}%',(bbox[0],bbox[1]-20),
                cv2.FONT_HERSHEY_PLAIN,3,(0,255,0),2)



    cTime=time.time()
    fps=1/(cTime-pTime)
    pTime=cTime
    cv2.putText(img,f'FPS: {int(fps)}',(20,70),cv2.FONT_HERSHEY_PLAIN,3,(0,255,0),2)



    cv2.imshow("Image",img)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break