| # 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 |