Update README.md
Browse files
README.md
CHANGED
|
@@ -5,4 +5,68 @@ base_model:
|
|
| 5 |
pipeline_tag: keypoint-detection
|
| 6 |
tags:
|
| 7 |
- sports
|
| 8 |
-
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
pipeline_tag: keypoint-detection
|
| 6 |
tags:
|
| 7 |
- sports
|
| 8 |
+
- swimming
|
| 9 |
+
- pose-estimation
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# SwimAnalysisPro: YOLO11-Pose for Side-View Swimming Analysis
|
| 13 |
+
# 側面游泳姿勢分析模型
|
| 14 |
+
|
| 15 |
+
## Model Description / 模型描述
|
| 16 |
+
|
| 17 |
+
This model is fine-tuned based on the Ultralytics YOLO11-pose architecture, specifically optimized for detecting and tracking swimmer poses from a side-view perspective. It is designed for technique analysis and biomechanical evaluation in aquatic environments.
|
| 18 |
+
|
| 19 |
+
本模型基於 Ultralytics YOLO11-pose 架構進行微調,專門針對側面視角的游泳者姿勢進行偵測與追蹤優化。適用於游泳技術分析以及水下或水面的生物力學評估。
|
| 20 |
+
|
| 21 |
+
## Training Dataset / 訓練資料集
|
| 22 |
+
|
| 23 |
+
The model was trained on a large-scale specialized dataset to ensure robustness across different styles.
|
| 24 |
+
|
| 25 |
+
本模型使用大規模專用資料集進行訓練,確保在不同泳姿下的穩定性。
|
| 26 |
+
|
| 27 |
+
- Total Images / 影像總數: 28,000
|
| 28 |
+
- Perspective / 視角: Side-view / 側面視角
|
| 29 |
+
- Stroke Coverage / 涵蓋泳姿: Butterfly, Backstroke, Breaststroke, and Freestyle / 蝶式、仰式、平式、自由式全涵蓋
|
| 30 |
+
|
| 31 |
+
## Keypoints Index / 關鍵點索引表
|
| 32 |
+
|
| 33 |
+
The model tracks 7 core human keypoints optimized for swimming stroke analysis:
|
| 34 |
+
|
| 35 |
+
本模型追蹤 7 個核心人體關鍵點,針對游泳動作分析進行優化:
|
| 36 |
+
|
| 37 |
+
| Index / 索引 | Keypoint (English) | 關鍵點 (中文) |
|
| 38 |
+
| --- | --- | --- |
|
| 39 |
+
| 0 | Head | 頭部 |
|
| 40 |
+
| 1 | Shoulder | 肩膀 |
|
| 41 |
+
| 2 | Elbow | 手肘 |
|
| 42 |
+
| 3 | Wrist | 手腕 |
|
| 43 |
+
| 4 | Hip | 髖部 |
|
| 44 |
+
| 5 | Knee | 膝蓋 |
|
| 45 |
+
| 6 | Ankle | 腳踝 |
|
| 46 |
+
|
| 47 |
+
## Performance / 效能指標
|
| 48 |
+
|
| 49 |
+
The evaluation metrics including mAP50 and mAP50-95 are provided in the charts below.
|
| 50 |
+
|
| 51 |
+
測試集上的效能數據(如 mAP50 與 mAP50-95)請參考下方圖表。
|
| 52 |
+
|
| 53 |
+

|
| 54 |
+
|
| 55 |
+
## How to Use / 如何使用
|
| 56 |
+
|
| 57 |
+
You can load and run this model directly using the Ultralytics Python package:
|
| 58 |
+
|
| 59 |
+
你可以使用 Ultralytics Python 套件直接載入並執行此模型:
|
| 60 |
+
|
| 61 |
+
```python
|
| 62 |
+
from ultralytics import YOLO
|
| 63 |
+
|
| 64 |
+
# Load the model
|
| 65 |
+
model = YOLO("your_username/your_model_name")
|
| 66 |
+
|
| 67 |
+
# Run inference
|
| 68 |
+
results = model.predict(source="swimming_video.mp4", save=True, conf=0.5)
|
| 69 |
+
|
| 70 |
+
# Process results
|
| 71 |
+
for result in results:
|
| 72 |
+
keypoints = result.keypoints.data
|