LibreReID-osnet

LibreReID-osnet

OSNet-AIN person re-identification weights for LibreYOLO's Deep OC-SORT tracker (model.track(source, tracker="deepocsort")). Downloaded automatically on first use to ~/.cache/libreyolo/reid/.

Files

osnet_ain_x0_25 is the LibreYOLO default. All files are plain PyTorch state dicts producing L2-normalized 512-d embeddings.

Provenance and license

• Network: OSNet-AIN, ported to LibreYOLO from Torchreid (MIT). The LibreYOLO port is state-dict compatible and bit-exact against upstream (tests/unit/test_reid.py).
• Weights: converted unchanged from the Torchreid model zoo multi-source (MS+D+C) OSNet-AIN checkpoints, released under the repository's MIT license. Conversion script: weights/convert_osnet_reid_weights.py in the LibreYOLO repository (strips the classifier head, keeps feature layers, verifies strict load).
• Training data: the upstream checkpoints were trained by the Torchreid authors on person re-identification research datasets (MSMT17, DukeMTMC-reID, CUHK03). Those datasets carry research-oriented terms; the weights themselves are distributed under MIT by the upstream author. Review your own use case if you deploy person re-identification in production.

Usage

from libreyolo import LibreYOLO

model = LibreYOLO("LibreYOLO9t.pt")
for result in model.track("video.mp4", tracker="deepocsort"):
    print(result.track_id)