闲来无聊,看到有的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