🚀 keremberke/yolov8n-csgo-player-detection
本项目基于ultralytics
库,使用YOLOv8n
模型实现了CSGO玩家检测。它能精准识别CSGO游戏画面中的CT和T玩家及其头部,在目标检测任务中表现出色,mAP@0.5(box)达到了0.84441。
🚀 快速开始
安装依赖
安装 ultralyticsplus:
pip install ultralyticsplus==0.0.23 ultralytics==8.0.21
加载模型并进行预测
from ultralyticsplus import YOLO, render_result
model = YOLO('keremberke/yolov8n-csgo-player-detection')
model.overrides['conf'] = 0.25
model.overrides['iou'] = 0.45
model.overrides['agnostic_nms'] = False
model.overrides['max_det'] = 1000
image = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
results = model.predict(image)
print(results[0].boxes)
render = render_result(model=model, image=image, result=results[0])
render.show()
更多模型请访问:awesome-yolov8-models
✨ 主要特性
- 精准检测:能够准确识别CSGO游戏画面中的CT和T玩家及其头部。
- 易于使用:借助
ultralyticsplus
库,简单的几行代码即可完成模型加载和预测。
📦 安装指南
安装 ultralyticsplus:
pip install ultralyticsplus==0.0.23 ultralytics==8.0.21
💻 使用示例
基础用法
from ultralyticsplus import YOLO, render_result
model = YOLO('keremberke/yolov8n-csgo-player-detection')
model.overrides['conf'] = 0.25
model.overrides['iou'] = 0.45
model.overrides['agnostic_nms'] = False
model.overrides['max_det'] = 1000
image = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
results = model.predict(image)
print(results[0].boxes)
render = render_result(model=model, image=image, result=results[0])
render.show()
📚 详细文档
支持的标签
['ct', 'cthead', 't', 'thead']
模型指标
属性 |
详情 |
模型类型 |
keremberke/yolov8n-csgo-player-detection |
训练数据集 |
keremberke/csgo-object-detection |
mAP@0.5(box) |
0.84441 |