Python yolov8实时屏幕检测demo

import mss
import cv2
import numpy as np
from ultralytics import YOLO

model=YOLO("yolov8n.pt")

# 检测的窗口位置
area = {"left":540,"top":140,"width":800,"height":800}

with mss.mss() as sct:
    while True:
        areascrs = sct.grab(area)
        areascrs = np.array(areascrs)
        areascrs2 = cv2.cvtColor(areascrs,cv2.COLOR_BGR2RGB)
        res = model(areascrs2)
        annotated_img = res[0].plot()


        # 窗口显示
        cv2.namedWindow("game",cv2.WINDOW_NORMAL)
        cv2.resizeWindow("game",area['width']//2,area['height']//2)
        cv2.setWindowProperty("game",cv2.WND_PROP_TOPMOST,1)
        # cv2.imshow("game",areascrs)
        cv2.imshow("game", cv2.cvtColor(annotated_img,cv2.COLOR_BGR2RGB))

        if cv2.waitKey(2) !=-1:
            cv2.destroyAllWindows()
            break