Update README.md
Browse files
README.md
CHANGED
|
@@ -16,11 +16,11 @@ pipeline_tag: object-detection
|
|
| 16 |
library_name: ultralytics
|
| 17 |
---
|
| 18 |
|
| 19 |
-
# 🏍️
|
| 20 |
|
| 21 |
-
<div
|
| 22 |
|
| 23 |
-
|
| 24 |
|
| 25 |
[](https://github.com/johnamit/mgp-detect)
|
| 26 |
|
|
@@ -28,13 +28,13 @@ library_name: ultralytics
|
|
| 28 |
|
| 29 |
## Model Overview
|
| 30 |
|
| 31 |
-
This repository contains trained model weights for the [
|
| 32 |
|
| 33 |
### Included Weights
|
| 34 |
|
| 35 |
| File | Description | Size |
|
| 36 |
|------|-------------|------|
|
| 37 |
-
| `motogp_yolov8m_detector.pt` | Fine-tuned YOLOv8m for
|
| 38 |
| `dinov3_team_classifier.pkl` | Logistic Regression classifier trained on DINOv3 embeddings | ~35KB |
|
| 39 |
|
| 40 |
## Pipeline Architecture
|
|
@@ -95,25 +95,6 @@ git clone https://github.com/facebookresearch/dinov3.git
|
|
| 95 |
```
|
| 96 |
Download the ViT-S/16 pretrained weights from the [DINOv3 repo](https://github.com/facebookresearch/dinov3).
|
| 97 |
|
| 98 |
-
### 3. Inference Example
|
| 99 |
-
|
| 100 |
-
```python
|
| 101 |
-
from ultralytics import YOLO
|
| 102 |
-
import torch
|
| 103 |
-
import joblib
|
| 104 |
-
|
| 105 |
-
# Load detector
|
| 106 |
-
detector = YOLO(detector_path)
|
| 107 |
-
|
| 108 |
-
# Load classifier
|
| 109 |
-
classifier = joblib.load(classifier_path)
|
| 110 |
-
|
| 111 |
-
# Run detection
|
| 112 |
-
results = detector.predict(source="race_frame.jpg", conf=0.5)
|
| 113 |
-
|
| 114 |
-
# For each detection, extract DINOv3 features and classify
|
| 115 |
-
# See full pipeline: https://github.com/johnamit/mgp-detect
|
| 116 |
-
```
|
| 117 |
|
| 118 |
## Training Details
|
| 119 |
|
|
|
|
| 16 |
library_name: ultralytics
|
| 17 |
---
|
| 18 |
|
| 19 |
+
# 🏍️ MotoReID:
|
| 20 |
|
| 21 |
+
<div>
|
| 22 |
|
| 23 |
+
A deep learning pipeline for **MotoGP team detection, tracking, and re-identification** from race broadcast footage. This system combines **YOLOv8** for robust object detection with **DINOv3** (Vision Transformer) embeddings for semantic team classification. It addresses specific challenges in high-speed sports computer vision: persistent identity tracking across extreme occlusions, rapid camera cuts, and motion blur. This project is in active development.
|
| 24 |
|
| 25 |
[](https://github.com/johnamit/mgp-detect)
|
| 26 |
|
|
|
|
| 28 |
|
| 29 |
## Model Overview
|
| 30 |
|
| 31 |
+
This repository contains trained model weights for the [MotoReID](https://github.com/johnamit/mgp-detect) project.
|
| 32 |
|
| 33 |
### Included Weights
|
| 34 |
|
| 35 |
| File | Description | Size |
|
| 36 |
|------|-------------|------|
|
| 37 |
+
| `motogp_yolov8m_detector.pt` | Fine-tuned YOLOv8m for MotoGP prototype (bike) detection | ~50MB |
|
| 38 |
| `dinov3_team_classifier.pkl` | Logistic Regression classifier trained on DINOv3 embeddings | ~35KB |
|
| 39 |
|
| 40 |
## Pipeline Architecture
|
|
|
|
| 95 |
```
|
| 96 |
Download the ViT-S/16 pretrained weights from the [DINOv3 repo](https://github.com/facebookresearch/dinov3).
|
| 97 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 98 |
|
| 99 |
## Training Details
|
| 100 |
|