介绍
本项目为CF网游,人物角色识别定位模型,采用yolov5框架实现,仅供学习研究使用。
模型效果展示
开发思路与流程
思路
CV领域的目标检测模型已经非常成熟,特别是yolov5系列 为什么不是用v6、v7?因为坑很多,而且效果不尽然好。
FPS游戏中,主要以击败敌方角色为目标,影响比赛成绩最主要的因素就是玩家的枪法。 使用深度学习的模型就可以对敌方角色进行准确定位,从而帮助玩家进行瞄准。而yolov5的预训练模型对于人物的识别已经很成熟。这时候只需要准备一定的数据样本,就可以实现不错效果。
流程
sample_util/collecting_data.pyF~sample/data/QFsample/data/BW
样本标注
使用开源数据标注工具labelImg
models/yolov5n.yaml
data\cf.yaml../cfdata
train.py--epochs300--workers2-4
yolov5-7.0/runs/train/exp/bast.pt
环境配置
系统要求:windows10+GPU
python>=3.8
安装依赖包
cd yolov5-7.0 进入yolov5-7.0目录
pip install -r requirements.txt
模型测试
测试模型效果,判断是否正常运行
python detect.py
yolov5-7.0/runs/detect/
在真实游戏中使用
运行predict.py文件
python predict.py
会连续截屏,并在定位到目标后,等待“鼠标右键”点击,如果捕获到右键,则会自动移动鼠标到目标位置