目录
1. 对屏幕固定区域的检测
实时检测电脑屏幕的某一个区域,对这部分区域进行检测
1.1 代码
代码实现比较简单,利用ImageGrab 抓取指定的区域,就可以实现单张图像的检测。
当加入了循环后,就可以实现一个实时的区域检测
因为当时训练这个网络的时候,模型是单通道输入的,所以这里将输入的图像更改为灰度图
import torch
import numpy as np
import cv2
from torchvision.transforms import transforms
from torchvision.models import detection
from PIL import ImageGrab # 抓取屏幕的文件
# 预处理
transformer = transforms.Compose([transforms.ToTensor()])
# 运行设备
DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
# 类名转换
id_to_class = {'1':'A'}
# 加载网络模型
model = detect