# apex_enemy_detect

> [English version](README.md)

Модели YOLOv8 для обнаружения врагов на записях геймплея Apex Legends. Два варианта: nano (скорость) и medium (точность).

[Apex Enemy Detect Demo](https://github.com/PSImera/apex_enemy_detect_demo) — инструмент для анализа видео геймплея, обнаружения врагов и исправления растянутого разрешения.

## Модели

| | **YOLOv8n (nano)** | **YOLOv8m (medium)** |
|---|---|---|
| Файл | `apex_detect_v8n_v2.1.pt` | `apex_detect_v8m_v2.1.pt` |
| Параметры | ~3.2M | ~25.9M |
| Precision | 0.932 | 0.943 |
| Recall | 0.877 | 0.883 |
| mAP@50 | 0.930 | 0.938 |
| mAP@50-95 | 0.756 | 0.796 |
| Лучше для | Скорость / слабый GPU | Точность |

## Использование

```python
from ultralytics import YOLO

model = YOLO("apex_detect_v8m_v2.1.pt")
results = model.predict("frame.jpg", conf=0.35, iou=0.5, imgsz=640)
```

Или автоматически через приложение [Apex Enemy Detect Demo](https://github.com/PSImera/apex_enemy_detect_demo) — модели загружаются отсюда при первом запуске.

## Параметры обучения

| Параметр | Значение |
|---|---|
| Датасет | `apex_detect_v2_p1_converted` |
| Эпохи | 200 (patience 100) |
| Batch size | 16 |
| Image size | 640×640 |
| Оптимизатор | AdamW (lr=0.001) |
| Аугментации | HSV (S/V ±0.3), горизонтальный флип (p=0.5), random erasing (p=0.4) |

Дообучены на основе весов COCO от Ultralytics на кастомном датасете врагов Apex Legends.

## Кривые обучения

<table>
<tr><th>YOLOv8n (nano)</th><th>YOLOv8m (medium)</th></tr>
<tr>
<td><img src="run_v8n_v2.1/results.png"></td>
<td><img src="run_v8m_v2.1/results.png"></td>
</tr>
</table>

## Кривые Precision-Recall

<table>
<tr><th>YOLOv8n (nano)</th><th>YOLOv8m (medium)</th></tr>
<tr>
<td><img src="run_v8n_v2.1/BoxPR_curve.png"></td>
<td><img src="run_v8m_v2.1/BoxPR_curve.png"></td>
</tr>
</table>