Binssin commited on
Commit
81697b7
·
verified ·
1 Parent(s): a0bb9d7

Upload 3 files

Browse files
Files changed (3) hide show
  1. README.md +62 -0
  2. last.pt +3 -0
  3. 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
+ - 在複雜背景或光線條件下可能表現不佳