YOLO即時人臉偵測opencv python實作

[python 請進]

import cv2
from ultralytics import YOLO
model=YOLO('yolov12n-face.pt') 
cap=cv2.VideoCapture(0, cv2.CAP_DSHOW)
cap.set(3, 1280) #cv2.CAP_PROP_FRAME_WIDTH
cap.set(4, 720) # cv2.CAP_PROP_FRAME_HEIGHT
cap.isOpened()
 
cnt=0
while True:
    ret, frame = cap.read()
    if not ret:
        break
    # 進行推論 (自動呼叫 PyTorch 底層運算)
    res=model(frame, conf=0.5, verbose=True)
    # 取得繪製好邊界框與關鍵點的影像
    im=res[0].plot()
    cv2.imshow("YOLO", im)
    k=cv2.waitKey(1)
    if k>0: # 按鍵 寫圖檔
        cv2.imwrite(f"{cnt}.jpg", im)
        print(f"{cnt}.jpg 寫進圖檔 ")
        cnt+=1
    if k==27: break
cap.release()
cv2.destroyAllWindows()