Hugging Face's logo

qualcomm
/
OSNet

PyTorch
backbone
real_time
android
video-object-tracking
Model card Files
xet
 Community
OSNet / README.md
qaihm-bot's picture
qaihm-bot
v0.63.0
52b4833 verified
preview code
|
Raw History Blame Contribute Delete
12.2 kB
metadata 
library_name: pytorch
license: mit
tags:
  - backbone
  - real_time
  - android
pipeline_tag: video-object-tracking

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.27.1 Download
ONNX w8a8 Universal QAIRT 2.50, ONNX Runtime 1.27.1 Download
QNN_DLC float 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.459 ms 0 - 56 MB NPU
OSNet ONNX float Snapdragon® 8 Elite For Galaxy Mobile 0.487 ms 0 - 54 MB NPU
OSNet ONNX float Snapdragon® X2 Elite 0.497 ms 1 - 1 MB NPU
OSNet ONNX float Snapdragon® X Elite 0.931 ms 0 - 0 MB NPU
OSNet ONNX float Snapdragon® 8 Gen 3 Mobile 0.607 ms 0 - 71 MB NPU
OSNet ONNX float Snapdragon® 8 Gen 1 Mobile 1.57 ms 0 - 71 MB NPU
OSNet ONNX float Qualcomm® Dragonwing™ IQ-8275 1.404 ms 0 - 4 MB NPU
OSNet ONNX float Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.899 ms 0 - 54 MB NPU
OSNet ONNX float Qualcomm® QCS8450 1.57 ms 0 - 71 MB NPU
OSNet ONNX float Qualcomm® Dragonwing™ IQ-9075 1.394 ms 0 - 4 MB NPU
OSNet ONNX float Qualcomm® Dragonwing™ IQ-X7181 0.931 ms 0 - 0 MB NPU
OSNet ONNX float Qualcomm® Dragonwing™ Q-8750 0.487 ms 0 - 54 MB NPU
OSNet ONNX w8a8 Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.281 ms 0 - 64 MB NPU
OSNet ONNX w8a8 Snapdragon® 8 Elite For Galaxy Mobile 0.326 ms 0 - 61 MB NPU
OSNet ONNX w8a8 Snapdragon® X2 Elite 0.27 ms 1 - 1 MB NPU
OSNet ONNX w8a8 Snapdragon® X Elite 0.548 ms 0 - 0 MB NPU
OSNet ONNX w8a8 Snapdragon® 8 Gen 3 Mobile 0.386 ms 0 - 77 MB NPU
OSNet ONNX w8a8 Snapdragon® 8 Gen 1 Mobile 0.68 ms 0 - 80 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ QCS6490 1.446 ms 0 - 3 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ IQ-8275 0.633 ms 0 - 4 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.548 ms 0 - 5 MB NPU
OSNet ONNX w8a8 Qualcomm® QCS8450 0.68 ms 0 - 80 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ IQ-9075 0.683 ms 0 - 3 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ IQ-X7181 0.548 ms 0 - 0 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ Q-6690 2.875 ms 0 - 66 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ Q-7790 0.586 ms 0 - 63 MB NPU
OSNet ONNX w8a8 Qualcomm® Dragonwing™ Q-8750 0.326 ms 0 - 61 MB NPU
OSNet ONNX w8a8 Snapdragon® 7 Gen 4 Mobile 0.586 ms 0 - 63 MB NPU
OSNet QNN_DLC float Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.433 ms 0 - 51 MB NPU
OSNet QNN_DLC float Snapdragon® 8 Elite For Galaxy Mobile 0.529 ms 0 - 48 MB NPU
OSNet QNN_DLC float Snapdragon® X2 Elite 0.633 ms 0 - 0 MB NPU
OSNet QNN_DLC float Snapdragon® X Elite 1.237 ms 0 - 0 MB NPU
OSNet QNN_DLC float Snapdragon® 8 Gen 3 Mobile 0.694 ms 0 - 62 MB NPU
OSNet QNN_DLC float Snapdragon® 8 Gen 1 Mobile 1.978 ms 0 - 65 MB NPU
OSNet QNN_DLC float Qualcomm® Dragonwing™ IQ-8275 1.438 ms 0 - 4 MB NPU
OSNet QNN_DLC float Qualcomm® Dragonwing™ QCS8550 (Proxy) 1.046 ms 0 - 2 MB NPU
OSNet QNN_DLC float Qualcomm® SA8775P 1.632 ms 0 - 52 MB NPU
OSNet QNN_DLC float Qualcomm® SA8650P 1.632 ms 0 - 52 MB NPU
OSNet QNN_DLC float Qualcomm® SA8255P 1.632 ms 0 - 52 MB NPU
OSNet QNN_DLC float Qualcomm® QCS8450 1.978 ms 0 - 65 MB NPU
OSNet QNN_DLC float Qualcomm® Dragonwing™ IQ-9075 1.503 ms 2 - 5 MB NPU
OSNet QNN_DLC float Qualcomm® Dragonwing™ IQ-X7181 1.237 ms 0 - 0 MB NPU
OSNet QNN_DLC float Qualcomm® Dragonwing™ Q-8750 0.529 ms 0 - 48 MB NPU
OSNet QNN_DLC float Qualcomm® SA7255P 4.523 ms 0 - 48 MB NPU
OSNet QNN_DLC float Qualcomm® SA8295P 2.073 ms 0 - 43 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.225 ms 0 - 48 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® 8 Elite For Galaxy Mobile 0.271 ms 0 - 46 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® X2 Elite 0.328 ms 0 - 0 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® X Elite 0.654 ms 0 - 0 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® 8 Gen 3 Mobile 0.366 ms 0 - 61 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® 8 Gen 1 Mobile 0.684 ms 0 - 65 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ QCS6490 1.438 ms 0 - 2 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ IQ-8275 0.534 ms 0 - 3 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.524 ms 0 - 2 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® SA8775P 0.725 ms 0 - 52 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® SA8650P 0.725 ms 0 - 52 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® SA8255P 0.725 ms 0 - 52 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® QCS8450 0.684 ms 0 - 65 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ IQ-9075 0.618 ms 0 - 2 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ IQ-X7181 0.654 ms 0 - 0 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ Q-6690 2.858 ms 0 - 47 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ Q-7790 0.534 ms 0 - 48 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® Dragonwing™ Q-8750 0.271 ms 0 - 46 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® SA7255P 1.223 ms 0 - 48 MB NPU
OSNet QNN_DLC w8a8 Qualcomm® SA8295P 0.851 ms 0 - 47 MB NPU
OSNet QNN_DLC w8a8 Snapdragon® 7 Gen 4 Mobile 0.534 ms 0 - 48 MB NPU
OSNet TFLITE float Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.438 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.697 ms 0 - 63 MB NPU
OSNet TFLITE float Snapdragon® 8 Gen 1 Mobile 1.972 ms 0 - 66 MB NPU
OSNet TFLITE float Qualcomm® Dragonwing™ IQ-8275 1.473 ms 0 - 8 MB NPU
OSNet TFLITE float Qualcomm® Dragonwing™ QCS8550 (Proxy) 1.095 ms 0 - 3 MB NPU
OSNet TFLITE float Qualcomm® QCS8450 1.972 ms 0 - 66 MB NPU
OSNet TFLITE float Qualcomm® Dragonwing™ IQ-9075 1.542 ms 0 - 8 MB NPU
OSNet TFLITE float Qualcomm® Dragonwing™ Q-8750 0.532 ms 0 - 48 MB NPU
OSNet TFLITE float Qualcomm® SA7255P 13.319 ms 0 - 23 MB GPU
OSNet TFLITE float Qualcomm® SA8295P 2.114 ms 0 - 42 MB NPU
OSNet TFLITE w8a8 Snapdragon® 8 Elite Gen 5 For Galaxy Mobile 0.208 ms 0 - 46 MB NPU
OSNet TFLITE w8a8 Snapdragon® 8 Elite For Galaxy Mobile 0.248 ms 0 - 43 MB NPU
OSNet TFLITE w8a8 Snapdragon® 8 Gen 3 Mobile 0.307 ms 0 - 57 MB NPU
OSNet TFLITE w8a8 Snapdragon® 8 Gen 1 Mobile 0.583 ms 0 - 63 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ QCS6490 2.243 ms 0 - 12 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ IQ-8275 0.475 ms 0 - 6 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.438 ms 0 - 2 MB NPU
OSNet TFLITE w8a8 Qualcomm® QCS8450 0.583 ms 0 - 63 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ IQ-9075 0.56 ms 0 - 5 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ Q-6690 3.696 ms 0 - 44 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ Q-7790 1.026 ms 0 - 46 MB NPU
OSNet TFLITE w8a8 Qualcomm® Dragonwing™ Q-8750 0.248 ms 0 - 43 MB NPU
OSNet TFLITE w8a8 Qualcomm® SA7255P 1.085 ms 0 - 45 MB NPU
OSNet TFLITE w8a8 Qualcomm® SA8295P 0.806 ms 0 - 44 MB NPU
OSNet TFLITE w8a8 Snapdragon® 7 Gen 4 Mobile 1.026 ms 0 - 46 MB NPU

License

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

References

Community