1-12. 洗濯槽内のティッシュを画像処理で検出してみた（③リアルタイム検出編）

import cv2
from ultralytics import YOLO

#学習済みモデル
model = YOLO('last.pt')

#カメラ起動
cap = cv2.VideoCapture(0)
if not cap.isOpened():
    print("Don't open camera")
    exit()

while True:
    #キャプチャ
    ret, frame = cap.read()
    if not ret:
        print("Don't get frame. exit")
        break

    #モデルで検出
    result = model(frame)

    #結果の画像を表示
    img_annotated = result[0].plot()
    cv2.imshow('Camera', img_annotated)

    #qを押すと終了
    if cv2.waitKey(1) == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()