YOLOv7实时推理

原创

已于 2024-07-17 10:38:32 修改 · 359 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#numpy #YOLO

于 2024-07-17 10:30:35 首次发布

YOLOv7实时推理

1.引入库

代码如下（示例）：

import threading
import time
import torch
import cv2
import numpy as np
import random
import win32con
import win32gui
import win32ui
from utils.general import non_max_suppression
from models.experimental import attempt_load

2.主要代码

代码如下（示例）：

classes = ["classes1", "classes2", "classes3", "classes4"] #标签类别
# colors = [[random.randint(0, 255) for _ in range(3)] for _ in classes]  # 随机颜色

def grab_screen(region=None):
    hwin = win32gui.GetDesktopWindow()
    left, top, x2, y2 = region
    width = x2 - left + 1
    height = y2 - top + 1

    hwindc = win32gui.GetWindowDC(hwin)
    srcdc = win32ui.CreateDCFromHandle(hwindc)
    memdc = srcdc.CreateCompatibleDC()
    bmp = win32ui.CreateBitmap()
    bmp.CreateCompatibleBitmap(srcdc, width, height)
    memdc.SelectObject(bmp)
    memdc.BitBlt((0, 0), (width, height), srcdc, (left, top), win32con.SRCCOPY)

    signedIntsArray = bmp.GetBitmapBits(True)
    img = np.fromstring(signedIntsArray, dtype