FPS射击游戏 yolov5 tensorrt 自瞄源码(开源) 只需10分钟部署trt环境

AI 是拿来用的,不是用来坑小白钱的,但是总有“穷疯了的人”会拿来到处卖

框架说明:

源码github :GitHub - xiao809407842/yolov5-tensorrt-aimbot: FPS game AI aimassist framework

需要有一点点python基础,完全没有编程基础建议学半个月再来

此ai框架支持.pt 和 .trt 两种模型(yolov5 6.1类里的DetectMultiBackend函数自带自动识别并加载各种模型,本来还支持onnx的,但是我不想再下载TensorFlow了),Pt模型应该是人人都能用,trt需要根据自己的显卡编译一次(编译方法最后有官方链接)。

模型输入维度是640*640 ,你可以换自己的权重,如果用自己的网络模型输入尺寸不是640*640,修改imgz = (640, 640)即可,模型需要fp16,如果不是fp16模型关闭half即可(half = False),尽量使用半精度模型,速度更快,而且精度没什么丢失。

修改瞄准的参数都在config.txt文档里,config.txt文档不要改顺序,也不加空行(主要是读取txt的函数写得有点麻瓜)

main.py注释掉的如 # ser.write(f'km.move({pid_x},{pid_y})\r\n'.encode('utf-8')) 等语句是我用的kmbox位移鼠标的语句。你都拿到相对位移坐标了,自己加一个鼠标位移就行

安装说明(一定按以下顺序安装环境依赖包

1. 安装anaconda 或者 mini conda

2. conda 创建虚拟环境

3.启动虚拟环境

conda activate

  1. 如果是中国大陆用户切换清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --set show_channel_urls yes

5.安装cuda、cudnn 必须先安装cuda 和cudnn)

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

6.安装tensorrt:

去英伟达官网下载TensorRT-8.2.1.8.Windows10.x86_64.安装包(只要是支持的版本都可以),解压后cd切换到 TensorRT-8.2.1.8\python文件夹, 使用pip install命令安装对应python版本的.whl文件(cp39对应python3.9,cp36对应python3.6)

示例:  pip install tensorrt-8.2.1.8-cp39-none-win_amd64.whl

7.安装yolov5 version6.1依赖:

pip install -r requirements.txt

8.完成上面安装后,最后安装win32 库(可能会报错,网上有解决方法,很容易解决):

pip install pypiwin32

或者pip install pywin32

9.运行main.py 

可以打开apex_test.jpg的图片,坐面拖动图片测试,正常情况扫识别到目标应该会打印坐标

补充

Pt模型转trt模型 ,yolov5 github官方有详细说明:

TFLite, ONNX, CoreML, TensorRT Export · Issue #251 · ultralytics/yolov5 · GitHub   

下载yolov5 6.1,用上面的环境编译即可

开源by:山高水长流

你可能感兴趣的:(yolov5,射击游戏,深度学习,python,游戏,人工智能,pytorch)