Upload 3 files
Browse files- README.md +62 -0
- last.pt +3 -0
- model_card.md +62 -0
README.md
ADDED
|
@@ -0,0 +1,62 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Pool Ball Detector
|
| 2 |
+
|
| 3 |
+
基於 YOLOv8 的撞球檢測模型。
|
| 4 |
+
|
| 5 |
+
## 安裝
|
| 6 |
+
|
| 7 |
+
```bash
|
| 8 |
+
pip install ultralytics
|
| 9 |
+
```
|
| 10 |
+
|
| 11 |
+
## 使用方法
|
| 12 |
+
|
| 13 |
+
1. 載入模型:
|
| 14 |
+
```python
|
| 15 |
+
from ultralytics import YOLO
|
| 16 |
+
model = YOLO('pool-ball-detector.pt')
|
| 17 |
+
```
|
| 18 |
+
|
| 19 |
+
2. 進行預測:
|
| 20 |
+
```python
|
| 21 |
+
results = model.predict('path/to/image.jpg')
|
| 22 |
+
```
|
| 23 |
+
|
| 24 |
+
## 訓練
|
| 25 |
+
|
| 26 |
+
```python
|
| 27 |
+
from ultralytics import YOLO
|
| 28 |
+
|
| 29 |
+
# 載入模型
|
| 30 |
+
model = YOLO('yolov8n.pt')
|
| 31 |
+
|
| 32 |
+
# 訓練
|
| 33 |
+
model.train(
|
| 34 |
+
data='data/dataset.yaml',
|
| 35 |
+
epochs=120,
|
| 36 |
+
imgsz=640,
|
| 37 |
+
batch=128,
|
| 38 |
+
device='0'
|
| 39 |
+
)
|
| 40 |
+
```
|
| 41 |
+
|
| 42 |
+
## 數據集結構
|
| 43 |
+
|
| 44 |
+
```
|
| 45 |
+
data/
|
| 46 |
+
├── dataset.yaml
|
| 47 |
+
├── images/
|
| 48 |
+
│ ├── train/
|
| 49 |
+
│ └── val/
|
| 50 |
+
└── labels/
|
| 51 |
+
├── train/
|
| 52 |
+
└── val/
|
| 53 |
+
```
|
| 54 |
+
|
| 55 |
+
## 模型性能
|
| 56 |
+
|
| 57 |
+
- mAP50: 0.931
|
| 58 |
+
- mAP50-95: 0.581
|
| 59 |
+
|
| 60 |
+
## 授權
|
| 61 |
+
|
| 62 |
+
MIT License
|
last.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7f3079fdf942b7f2df062a70a63c0e74d34b8fa8b200d06236cdc684ac144b6e
|
| 3 |
+
size 6244387
|
model_card.md
ADDED
|
@@ -0,0 +1,62 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: zh
|
| 3 |
+
tags:
|
| 4 |
+
- computer-vision
|
| 5 |
+
- object-detection
|
| 6 |
+
- yolo
|
| 7 |
+
- pool-ball
|
| 8 |
+
license: mit
|
| 9 |
+
datasets:
|
| 10 |
+
- custom-pool-ball
|
| 11 |
+
metrics:
|
| 12 |
+
- mAP50
|
| 13 |
+
- mAP50-95
|
| 14 |
+
---
|
| 15 |
+
|
| 16 |
+
# Pool Ball Detector
|
| 17 |
+
|
| 18 |
+
這是一個基於 YOLOv8 的撞球檢測模型,用於識別和定位撞球。
|
| 19 |
+
|
| 20 |
+
## 模型詳情
|
| 21 |
+
|
| 22 |
+
- 模型架構:YOLOv8n
|
| 23 |
+
- 訓練數據:自定義撞球數據集
|
| 24 |
+
- 訓練輪數:120 epochs
|
| 25 |
+
- 圖像大小:640x640
|
| 26 |
+
- 批次大小:128
|
| 27 |
+
- 設備:NVIDIA GeForce RTX 3060
|
| 28 |
+
|
| 29 |
+
## 性能指標
|
| 30 |
+
|
| 31 |
+
- mAP50: 0.931
|
| 32 |
+
- mAP50-95: 0.581
|
| 33 |
+
|
| 34 |
+
## 使用方法
|
| 35 |
+
|
| 36 |
+
```python
|
| 37 |
+
from ultralytics import YOLO
|
| 38 |
+
|
| 39 |
+
# 載入模型
|
| 40 |
+
model = YOLO('pool-ball-detector.pt')
|
| 41 |
+
|
| 42 |
+
# 進行預測
|
| 43 |
+
results = model.predict('path/to/image.jpg')
|
| 44 |
+
```
|
| 45 |
+
|
| 46 |
+
## 訓練細節
|
| 47 |
+
|
| 48 |
+
- 使用 YOLOv8n 預訓練模型
|
| 49 |
+
- 使用自動混合精度訓練
|
| 50 |
+
- 使用早停策略
|
| 51 |
+
- 每 10 個 epoch 保存一次檢查點
|
| 52 |
+
|
| 53 |
+
## 數據集
|
| 54 |
+
|
| 55 |
+
自定義撞球數據集,包含:
|
| 56 |
+
- 訓練集:撞球圖像和對應的標註
|
| 57 |
+
- 驗證集:用於評估模型性能
|
| 58 |
+
|
| 59 |
+
## 限制
|
| 60 |
+
|
| 61 |
+
- 模型僅針對撞球檢測進行訓練
|
| 62 |
+
- 在複雜背景或光線條件下可能表現不佳
|