基于YOLOv5的王者荣耀目标检测4-拓展-对电脑屏幕进行实时检测
一、前言
上一篇博客:基于YOLOv5的王者荣耀目标检测3-简单地利用训练好的模型进行推理
二、正文
1、思路
无论是视频还是摄像头,YOLO的目标检测实际上都是对图像进行预测的因此对于电脑屏幕的实时检测。我们可以不断地截屏,将图片逐张推理,再把推理结果返回窗口。
过程分为三步:①截屏。此处可以使用MSS来进行实时的截屏;②使用YOLO进行推理。这里我们调用run()方法即可③展示推理结果。将推理结果展示到窗口上。
2、实践
2.1、detect.py结构
detect.py文件中主要有两个类,一个用于设置参数,一个用于进行推理。
2.2、使用MSS实时截屏并推理
3、效果展示
王者荣耀屏幕实时检测
4、总结
因为是截一张图推理一次,所以整个推理的过程比较卡顿,还不能像使用摄像头那样进行流畅地检测。
可能的改进方法:像摄像头一样通过视频流来输入图片,一次处理多张图片。
我的项目源码(不含数据集)