闲来无聊,看到有的AI大神通过图像识别玩fps游戏 ,检测头部或者其他肢体来完成射击,突然来了兴趣,想自己弄一个,于是乎,参照其中一个改了下,原版本是用的是YOLOv5-v5版本,经过自己修改,目前在yolov5-v6.0版本下面正常运行

原始版本  项目地址

再次精简后的版本  项目地址(推荐)

本机实测正常环境:

python 3.8.10
yolov5的v6.0
pywin32 302
cuda Build cuda_11.6.r11.6
pytorch 1.10.0+cu113
torchvision 0.11.1+cu113

硬件:

AMD 5800H
RTX3060
16G内存
三星980  1T固态

在上面的平台,平均检测到射击的时间为15-20 ms,大约一帧的时间

在 8G内存,gtx 1070上面也测试通过,平均检测到射击时间为60ms