Hugging Face's logo

qualcomm
/
OSNet

PyTorch
backbone
real_time
android
video-object-tracking
Model card Files
xet
 Community

YAML Metadata Warning:The pipeline tag "video-object-tracking" is not in the official list: text-classification, token-classification, table-question-answering, question-answering, zero-shot-classification, translation, summarization, feature-extraction, text-generation, fill-mask, sentence-similarity, text-to-speech, text-to-audio, automatic-speech-recognition, audio-to-audio, audio-classification, audio-text-to-text, voice-activity-detection, depth-estimation, image-classification, object-detection, image-segmentation, text-to-image, image-to-text, image-to-image, image-to-video, unconditional-image-generation, video-classification, reinforcement-learning, robotics, tabular-classification, tabular-regression, tabular-to-text, table-to-text, multiple-choice, text-ranking, text-retrieval, time-series-forecasting, text-to-video, image-text-to-text, image-text-to-image, image-text-to-video, visual-question-answering, document-question-answering, zero-shot-image-classification, graph-ml, mask-generation, zero-shot-object-detection, text-to-3d, image-to-3d, image-feature-extraction, video-text-to-text, keypoint-detection, visual-document-retrieval, any-to-any, video-to-video, other

OSNet: Optimized for Qualcomm Devices

OSNet is a compact convolutional architecture for person re-identification. It extracts discriminative embeddings that can be compared with cosine similarity for person matching across camera views.

This is based on the implementation of OSNet found here. This repository contains pre-exported model files optimized for Qualcomm® devices. You can use the Qualcomm® AI Hub Models library to export with custom configurations. More details on model performance across various devices, can be found here.

Qualcomm AI Hub Models uses Qualcomm AI Hub Workbench to compile, profile, and evaluate this model. Sign up to run these models on a hosted Qualcomm® device.

Getting Started

There are two ways to deploy this model on your device:

Option 1: Download Pre-Exported Models

Below are pre-exported model assets ready for deployment.

Runtime Precision Chipset SDK Versions Download
ONNX float Universal QAIRT 2.50, ONNX Runtime 1.30.0 Download
ONNX w8a16 Universal QAIRT 2.50, ONNX Runtime 1.30.0 Download
ONNX w8a8 Universal QAIRT 2.50, ONNX Runtime 1.30.0 Download
QNN_DLC float Universal QAIRT 2.50 Download
QNN_DLC w8a16 Universal QAIRT 2.50 Download
QNN_DLC w8a8 Universal QAIRT 2.50 Download
TFLITE float Universal QAIRT 2.50 Download
TFLITE w8a8 Universal QAIRT 2.50 Download

For more device-specific assets and performance metrics, visit OSNet on Qualcomm® AI Hub.

Option 2: Export with Custom Configurations

Use the Qualcomm® AI Hub Models Python library to compile and export the model with your own:

  • Custom weights (e.g., fine-tuned checkpoints)
  • Custom input shapes
  • Target device and runtime configurations

This option is ideal if you need to customize the model beyond the default configuration provided here.

See our repository for OSNet on GitHub for usage instructions.

Model Details

Model Type: Model_use_case.video_object_tracking

Model Stats:

  • Input resolution: 256x128
  • Model checkpoint: osnet_x1_0_market_256x128_amsgrad_ep150_stp60_lr0.0015_b64_fb10_softmax_labelsmooth_flip.pth
  • Model size (float): 8.69 MB
  • Model size (w8a8): 3.57 MB
  • Number of parameters: 2.68M

Performance Summary

Model Runtime Precision Chipset Inference Time (ms) Peak Memory Range (MB) Primary Compute Unit
OSNet ONNX float Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.463 ms 0 - 56 MB NPU
OSNet ONNX float Snapdragon® 8 Elite For Galaxy Mobile 0.486 ms 0 - 55 MB NPU
OSNet ONNX float Snapdragon® X2 Elite 0.498 ms 1 - 1 MB NPU
OSNet ONNX float Snapdragon® X Elite 0.929 ms 0 - 0 MB NPU
OSNet ONNX float Snapdragon® 8 Gen 3 Mobile 0.618 ms 0 - 71 MB NPU
OSNet ONNX float Snapdragon® 8 Gen 1 Mobile 1.558 ms 0 - 71 MB NPU
OSNet ONNX float Qualcomm® Dragonwing™ IQ-8275 1.4 ms 0 - 4 MB NPU
OSNet ONNX float Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.894 ms 0 - 6 MB NPU
OSNet ONNX float Qualcomm® QCS8450 1.558 ms 0 - 71 MB NPU
OSNet ONNX float Qualcomm® Dragonwing™ IQ-9075 1.404 ms 0 - 4 MB NPU
OSNet ONNX float Qualcomm® Dragonwing™ IQ-X7181 0.929 ms 0 - 0 MB NPU
OSNet ONNX float Qualcomm® Dragonwing™ Q-8750 0.486 ms 0 - 55 MB NPU
OSNet ONNX w8a16 Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.405 ms 0 - 66 MB NPU
OSNet ONNX w8a16 Snapdragon® 8 Elite For Galaxy Mobile 0.493 ms 0 - 69 MB NPU
OSNet ONNX w8a16 Snapdragon® X2 Elite 0.512 ms 1 - 1 MB NPU
OSNet ONNX w8a16 Snapdragon® X Elite 0.898 ms 0 - 0 MB NPU
OSNet ONNX w8a16 Snapdragon® 8 Gen 3 Mobile 0.63 ms 0 - 84 MB NPU
OSNet ONNX w8a16 Snapdragon® 8 Gen 1 Mobile 1.19 ms 0 - 87 MB NPU
OSNet ONNX w8a16 Qualcomm® Dragonwing™ QCS6490 2.415 ms 0 - 3 MB NPU
OSNet ONNX w8a16 Qualcomm® Dragonwing™ IQ-8275 0.985 ms 0 - 4 MB NPU
OSNet ONNX w8a16 Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.869 ms 0 - 8 MB NPU
OSNet ONNX w8a16 Qualcomm® QCS8450 1.19 ms 0 - 87 MB NPU
OSNet ONNX w8a16 Qualcomm® Dragonwing™ IQ-9075 1.017 ms 0 - 3 MB NPU
OSNet ONNX w8a16 Qualcomm® Dragonwing™ IQ-X7181 0.898 ms 0 - 0 MB NPU
OSNet ONNX w8a16 Qualcomm® Dragonwing™ Q-6690 4.87 ms 0 - 175 MB NPU
OSNet ONNX w8a16 Qualcomm® Dragonwing™ Q-7790 0.953 ms 0 - 68 MB NPU
OSNet ONNX w8a16 Qualcomm® Dragonwing™ Q-8750 0.493 ms 0 - 69 MB NPU
OSNet ONNX w8a16 Snapdragon® 7 Gen 4 Mobile 0.953 ms 0 - 68 MB NPU
OSNet ONNX w8a8 Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.297 ms 0 - 65 MB NPU
OSNet ONNX w8a8 Snapdragon® 8 Elite For Galaxy Mobile 0.342 ms 0 - 69 MB NPU
OSNet ONNX w8a8 Snapdragon® X2 Elite 0.284 ms 1 - 1 MB NPU
OSNet ONNX w8a8 Snapdragon® X Elite 0.576 ms 0 - 0 MB NPU
OSNet ONNX w8a8 Snapdragon® 8 Gen 3 Mobile 0.393 ms 0 - 81 MB NPU
OSNet ONNX w8a8 Snapdragon® 8 Gen 1 Mobile 0.712 ms 0 - 83 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ QCS6490 1.88 ms 0 - 3 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ IQ-8275 0.646 ms 0 - 4 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.567 ms 0 - 5 MB NPU
OSNet ONNX w8a8 Qualcomm® QCS8450 0.712 ms 0 - 83 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ IQ-9075 0.728 ms 0 - 3 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ IQ-X7181 0.576 ms 0 - 0 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ Q-6690 2.876 ms 0 - 69 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ Q-7790 0.608 ms 0 - 65 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ Q-8750 0.342 ms 0 - 69 MB NPU
OSNet ONNX w8a8 Snapdragon® 7 Gen 4 Mobile 0.608 ms 0 - 65 MB NPU
OSNet QNN_DLC float Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.436 ms 0 - 50 MB NPU
OSNet QNN_DLC float Snapdragon® 8 Elite For Galaxy Mobile 0.53 ms 0 - 48 MB NPU
OSNet QNN_DLC float Snapdragon® X2 Elite 0.645 ms 0 - 0 MB NPU
OSNet QNN_DLC float Snapdragon® X Elite 1.22 ms 0 - 0 MB NPU
OSNet QNN_DLC float Snapdragon® 8 Gen 3 Mobile 0.697 ms 0 - 62 MB NPU
OSNet QNN_DLC float Snapdragon® 8 Gen 1 Mobile 1.981 ms 0 - 65 MB NPU
OSNet QNN_DLC float Qualcomm® Dragonwing™ IQ-8275 1.44 ms 0 - 4 MB NPU
OSNet QNN_DLC float Qualcomm® Dragonwing™ QCS8550 (Proxy) 1.036 ms 0 - 3 MB NPU
OSNet QNN_DLC float Qualcomm® SA8775P 1.649 ms 0 - 52 MB NPU
OSNet QNN_DLC float Qualcomm® SA8650P 1.649 ms 0 - 52 MB NPU
OSNet QNN_DLC float Qualcomm® SA8255P 1.649 ms 0 - 52 MB NPU
OSNet QNN_DLC float Qualcomm® QCS8450 1.981 ms 0 - 65 MB NPU
OSNet QNN_DLC float Qualcomm® Dragonwing™ IQ-9075 1.499 ms 0 - 3 MB NPU
OSNet QNN_DLC float Qualcomm® Dragonwing™ IQ-X7181 1.22 ms 0 - 0 MB NPU
OSNet QNN_DLC float Qualcomm® Dragonwing™ Q-8750 0.53 ms 0 - 48 MB NPU
OSNet QNN_DLC float Qualcomm® SA7255P 4.526 ms 0 - 48 MB NPU
OSNet QNN_DLC float Qualcomm® SA8295P 2.064 ms 0 - 43 MB NPU
OSNet QNN_DLC w8a16 Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.359 ms 0 - 50 MB NPU
OSNet QNN_DLC w8a16 Snapdragon® 8 Elite For Galaxy Mobile 0.457 ms 0 - 47 MB NPU
OSNet QNN_DLC w8a16 Snapdragon® X2 Elite 0.576 ms 0 - 0 MB NPU
OSNet QNN_DLC w8a16 Snapdragon® X Elite 1.045 ms 0 - 0 MB NPU
OSNet QNN_DLC w8a16 Snapdragon® 8 Gen 3 Mobile 0.632 ms 0 - 64 MB NPU
OSNet QNN_DLC w8a16 Snapdragon® 8 Gen 1 Mobile 1.203 ms 0 - 70 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® Dragonwing™ QCS6490 2.633 ms 2 - 4 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® Dragonwing™ IQ-8275 0.94 ms 0 - 3 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.894 ms 0 - 2 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® SA8775P 1.115 ms 0 - 55 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® SA8650P 1.115 ms 0 - 55 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® SA8255P 1.115 ms 0 - 55 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® QCS8450 1.203 ms 0 - 70 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® Dragonwing™ IQ-9075 0.999 ms 0 - 2 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® Dragonwing™ IQ-X7181 1.045 ms 0 - 0 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® Dragonwing™ Q-6690 5.162 ms 0 - 158 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® Dragonwing™ Q-7790 0.973 ms 0 - 51 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® Dragonwing™ Q-8750 0.457 ms 0 - 47 MB NPU
OSNet QNN_DLC w8a16 Qualcomm® SA8295P 1.493 ms 0 - 51 MB NPU
OSNet QNN_DLC w8a16 Snapdragon® 7 Gen 4 Mobile 0.973 ms 0 - 51 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.229 ms 0 - 49 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® 8 Elite For Galaxy Mobile 0.289 ms 0 - 52 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® X2 Elite 0.355 ms 0 - 0 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® X Elite 0.687 ms 0 - 0 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® 8 Gen 3 Mobile 0.367 ms 0 - 62 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® 8 Gen 1 Mobile 0.681 ms 0 - 67 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ QCS6490 1.426 ms 0 - 2 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ IQ-8275 0.593 ms 0 - 3 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.557 ms 0 - 10 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® SA8775P 0.745 ms 0 - 53 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® SA8650P 0.745 ms 0 - 53 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® SA8255P 0.745 ms 0 - 53 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® QCS8450 0.681 ms 0 - 67 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ IQ-9075 0.646 ms 0 - 2 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ IQ-X7181 0.687 ms 0 - 0 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ Q-6690 2.869 ms 0 - 47 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ Q-7790 0.562 ms 0 - 48 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ Q-8750 0.289 ms 0 - 52 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® SA8295P 0.902 ms 0 - 48 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® 7 Gen 4 Mobile 0.562 ms 0 - 48 MB NPU
OSNet TFLITE float Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.434 ms 0 - 49 MB NPU
OSNet TFLITE float Snapdragon® 8 Elite For Galaxy Mobile 0.532 ms 0 - 48 MB NPU
OSNet TFLITE float Snapdragon® 8 Gen 3 Mobile 0.711 ms 0 - 64 MB NPU
OSNet TFLITE float Snapdragon® 8 Gen 1 Mobile 1.987 ms 0 - 67 MB NPU
OSNet TFLITE float Qualcomm® Dragonwing™ IQ-8275 1.472 ms 0 - 8 MB NPU
OSNet TFLITE float Qualcomm® Dragonwing™ QCS8550 (Proxy) 1.093 ms 0 - 2 MB NPU
OSNet TFLITE float Qualcomm® QCS8450 1.987 ms 0 - 67 MB NPU
OSNet TFLITE float Qualcomm® Dragonwing™ IQ-9075 1.544 ms 0 - 8 MB NPU
OSNet TFLITE float Qualcomm® Dragonwing™ Q-8750 0.532 ms 0 - 48 MB NPU
OSNet TFLITE float Qualcomm® SA8295P 2.115 ms 0 - 43 MB NPU
OSNet TFLITE w8a8 Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.215 ms 0 - 46 MB NPU
OSNet TFLITE w8a8 Snapdragon® 8 Elite For Galaxy Mobile 0.262 ms 0 - 50 MB NPU
OSNet TFLITE w8a8 Snapdragon® 8 Gen 3 Mobile 0.315 ms 0 - 61 MB NPU
OSNet TFLITE w8a8 Snapdragon® 8 Gen 1 Mobile 0.609 ms 0 - 64 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ QCS6490 1.887 ms 0 - 12 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ IQ-8275 0.519 ms 0 - 6 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.462 ms 0 - 2 MB NPU
OSNet TFLITE w8a8 Qualcomm® QCS8450 0.609 ms 0 - 64 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ IQ-9075 0.585 ms 0 - 5 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ Q-6690 3.618 ms 0 - 45 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ Q-7790 0.995 ms 0 - 46 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ Q-8750 0.262 ms 0 - 50 MB NPU
OSNet TFLITE w8a8 Qualcomm® SA8295P 0.824 ms 0 - 44 MB NPU
OSNet TFLITE w8a8 Snapdragon® 7 Gen 4 Mobile 0.995 ms 0 - 46 MB NPU

License

  • The license for the original implementation of OSNet can be found here.

References

Community

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Paper for qualcomm/OSNet