# Pool Ball Detector

基於 YOLOv8 的撞球檢測模型。

## 安裝

```bash
pip install ultralytics
```

## 使用方法

1. 載入模型：
```python
from ultralytics import YOLO
model = YOLO('pool-ball-detector.pt')
```

2. 進行預測：
```python
results = model.predict('path/to/image.jpg')
```

## 訓練

```python
from ultralytics import YOLO

# 載入模型
model = YOLO('yolov8n.pt')

# 訓練
model.train(
    data='data/dataset.yaml',
    epochs=120,
    imgsz=640,
    batch=128,
    device='0'
)
```

## 數據集結構

```
data/
├── dataset.yaml
├── images/
│   ├── train/
│   └── val/
└── labels/
    ├── train/
    └── val/
```

## 模型性能

- mAP50: 0.931
- mAP50-95: 0.581

## 授權

MIT License