源码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位移鼠标的语句。你都拿到相对位移坐标了,自己加一个鼠标位移就行
conda activate
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
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
去英伟达官网下载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
pip install -r requirements.txt
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:山高水长流