Skip to content

limboy058/aimbot_csgo2

Repository files navigation

基于Python, YOLO模型, TensorRT的高性能CSGO2自瞄外挂

代码是一点一点手搓的,没有参考任何已有的项目。

您可以查看从demo1到demo5的不同版本之间的变化

1. 项目结构

image-20240624163353789

csgo2文件夹中是我自建的数据集

train_results 包含了我训练过的所有模型的训练结果

demo_win_pytorch_normal 包含了我的大部分工作相关的代码,其中包含了我逐步开发的5个版本,并在代码中有简要说明。其中demo4.py是完全可运行的

get_train_data是我用来在游戏中截图,并分割数据集的代码

test_files 中包括了一些可以帮助您检测环境是否安装完成的代码

2. 如何跑通代码?

如果你想尝试更高的性能,就需要配置更多的东西。

我提供了以下几个版本:

image-20240624162206961

demo_mac是mac平台上的运行代码,我不清楚如何在mac上移动鼠标,所以您需要写一点代码才可以体验完整功能

demo_win_pytorch_easy是win平台上的运行代码,只需要可以运行ultralytics的代码就可以使用全部功能,但速度一般

demo_win_pytorch_normal 需要配置DXGI库和安装罗技驱动,可以取得很好的性能

demo_win_tensorrt_fast 需要配置DXGI库和安装罗技驱动,进一步需要配置tensorrt可以取得最好的性能

demo_win_pytorch_yolov5 是 yolov5版本的代码,需要配置DXGI库和安装罗技驱动

demo_win_pytorch_yolov10 是 yolov10版本的代码,需要配置DXGI库和安装罗技驱动

以上,请您按需选择。如果在安装中遇到了问题,请提问。

1.demo_mac

如果你想要在mac上运行,请使用此文件中的内容.

建议使用纯净的环境,

①参考官网配置好YOLOv8环境,确保torch、cuda、cudnn等的匹配

②进入此文件夹pip安装requirements.txt,

③修改此文件夹中demo4.py中的todo,例如

image-20240624170717006

并添加移动鼠标的代码

即可运行。注意权重文件在demo_win_pytorch_normal中,我并没有复制两份

2. demo_win_pytorch_easy

推荐使用这个版本。使用mss截图,windows鼠标移动事件控制鼠标。性能大概为25帧率还算够用,配置简单

建议使用纯净的环境,

①参考官网配置好YOLOv8环境,确保torch、cuda、cudnn等的匹配

②进入此文件夹pip安装requirements.txt,

③修改此文件夹中demo4.py中的todo,例如

image-20240624170717006

即可运行。注意权重文件在demo_win_pytorch_normal中,我并没有复制两份

3. demo_win_pytorch_normal

您需要在python3.9下运行这些代码。因为DXGI.pyd的版本是3.9的

**注意!**在这个文件夹中,由于opencv_world451.dllopencv_world3416.dllLGS_9.02.65_x64_Logitech.exe过大,我无法上传至github,请从网盘下载并放置到此目录下。

image-20240625205224948

链接:https://pan.baidu.com/s/1qxnk2-O394in103lYk3aJQ?pwd=qzlz ,网盘链接里的压缩包包含:

image-20240625214210114

您可以使用test_files中的文件来测试环境是否安装成功

①使用纯净的环境,参考官网配置好YOLOv8环境

②进入此文件夹pip安装requirements.txt。

③卸载罗技鼠标驱动(如果有的话),并安装驱动LGS_9.02.65_x64_Logitech.exe,其大小大概为120MB,您可以在网盘下载

打开此软件,电脑重启后需要重新打开软件。

image-20240624172447127

软件页面如图。不显示这个白色鼠标是正常的,这是我的鼠标。。没有罗技鼠标也可以调用这个软件。

使用test_files中的logitech_test.py测试驱动是否安装成功,如果成功了应该在运行后看到鼠标指针移动。

如果在安装时遇到问题,可以参考网上相关安装教程以及我写的罗技鼠标驱动卸载_罗技驱动卸载-CSDN博客

④修改此文件夹中demo4.py中的todo,例如

image-20240624170717006

即可运行。

4. demo_win_pytorch_yolov5

①使用纯净的环境,参考官网配置好YOLOv5环境

②进入此文件夹, pip安装requirements.txt。

③卸载罗技鼠标驱动(如果有的话),并安装此驱动LGS_9.02.65_x64_Logitech.exe,并打开此软件。电脑重启后需要重新打开软件。

④修改此文件夹中demo4.py 时间原因我没有写todo,看着改就可以了

5.demo_win_pytorch_yolov10

①使用纯净的环境,参考官网配置好YOLOv10环境

②进入此文件夹pip安装requirements.txt。

③卸载罗技鼠标驱动(如果有的话),并安装此驱动LGS_9.02.65_x64_Logitech.exe,并打开此软件。电脑重启后需要重新打开软件。

④修改此文件夹中demo4.py 时间原因我没有写todo,看着改就可以了

6.demo_win_tensorrt_fast

强烈建议您使用与我完全一致的版本!我使用的环境是:

nvidia驱动支持的最高版本:CUDA Version: 12.3(命令行输入nvidim-smi可查看)

cuda安装版本:12.1 (命令行输入nvcc -V查看)

python版本:3.9.19

torch版本:

pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121

cudnn版本:

image-20240624180135054

cuDNN Archive | NVIDIA Developer

tensorRT版本:

image-20240624180250617

NVIDIA TensorRT 8.x Download | NVIDIA Developer

以上相关环境的安装可以参考网上的教程,但建议保持您安装的版本与我的一致

然后请参考此连接配置yolov8-tensorrt环境:

yolov8实战第三天——yolov8TensorRT部署(python推理)(保姆教学)_yolov8 tensorrt python-CSDN博客

确保罗技驱动与所有代码用到的库都安装完成后,导出pt文件为onnx文件,再导出为engine文件,engine文件是此处demo4.py的模型参数。

拷贝demo_win_tensorrt_fast文件夹与YOLOv8_tensorrt处于同一目录下,如下。

image-20240624180834143

我导出的engine文件放在了pt文件夹下,由于每个电脑上都不一样,所以我没有上传。大致的文件结构如图:

image-20240624180932151

略微修改此文件夹中demo4.py中的todo,即可运行.

3.代码技术细节

参考doc.pdf

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages