Upload README.md with huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# OSNet Weights for Trackers
|
| 2 |
+
|
| 3 |
+
Pre-trained OSNet x1.0 weights for use with the [trackers](https://github.com/roboflow/trackers) library.
|
| 4 |
+
|
| 5 |
+
## Files
|
| 6 |
+
|
| 7 |
+
| File | Description | Source |
|
| 8 |
+
|---|---|---|
|
| 9 |
+
| `osnet_x1_0_imagenet.pth` | OSNet x1.0 pretrained on ImageNet (general features) | [torchreid](https://github.com/KaiyangZhou/deep-person-reid) |
|
| 10 |
+
| `sports_model.pth.tar-60` | OSNet x1.0 fine-tuned on SportsMOT crops for person ReID | [Deep-EIoU](https://github.com/hsiangwei0903/Deep-EIoU) |
|
| 11 |
+
|
| 12 |
+
## Usage
|
| 13 |
+
|
| 14 |
+
```python
|
| 15 |
+
from trackers.appearance.osnet import OSNetExtractor
|
| 16 |
+
|
| 17 |
+
# ImageNet pretrained (general)
|
| 18 |
+
extractor = OSNetExtractor(weights="imagenet")
|
| 19 |
+
|
| 20 |
+
# SportsMOT fine-tuned (sports ReID — used by Deep HM-SORT paper)
|
| 21 |
+
extractor = OSNetExtractor(weights="sportsmot")
|
| 22 |
+
```
|
| 23 |
+
|
| 24 |
+
## References
|
| 25 |
+
|
| 26 |
+
- Zhou et al., *Omni-Scale Feature Learning for Person Re-Identification*, ICCV 2019
|
| 27 |
+
- Huang et al., *Deep-EIoU*, arXiv:2306.13074
|
| 28 |
+
- Gran-Henriksen et al., *Deep HM-SORT*, arXiv:2406.12081
|