Pool Ball Detector

基於 YOLOv8 的撞球檢測模型。

安裝

pip install ultralytics

使用方法

載入模型：

from ultralytics import YOLO
model = YOLO('pool-ball-detector.pt')

進行預測：

results = model.predict('path/to/image.jpg')

訓練

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