CoolFace
Modelpublic

qualcomm/PidNet

sourceHugging Facemitupdated 10d agoView on Hugging Face
3likes150downloads
Model Card

PidNet: Optimized for Qualcomm Devices

PIDNet (Proportional-Integral-Derivative Network) is a real-time semantic segmentation model based on PID controllers

This is based on the implementation of PidNet 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.

RuntimePrecisionChipsetSDK VersionsDownload
ONNXfloatUniversalQAIRT 2.45, ONNX Runtime 1.27.1Download
ONNXw8a8UniversalQAIRT 2.45, ONNX Runtime 1.27.1Download
QNN_DLCfloatUniversalQAIRT 2.45Download
QNN_DLCw8a8UniversalQAIRT 2.45Download
TFLITEfloatUniversalQAIRT 2.45Download
TFLITEw8a8UniversalQAIRT 2.45Download

For more device-specific assets and performance metrics, visit [PidNet on Qualcomm® AI Hub](https://aihub.qualcomm.com/models/pidnet).

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 PidNet on GitHub for usage instructions.

Model Details

Model Type: Modelusecase.semantic_segmentation

Model Stats:

  • Inference latency: RealTime
  • Input resolution: 1024x2048
  • Model checkpoint: PIDNetSCityscapes_val.pt
  • Model size (float): 29.1 MB
  • Model size (w8a8): 8.02 MB
  • Number of output classes: 19
  • Number of parameters: 8.06M

Performance Summary

ModelRuntimePrecisionChipsetInference Time (ms)Peak Memory Range (MB)Primary Compute Unit
PidNetONNXfloatSnapdragon® X2 Elite13.743 ms24 - 24 MBNPU
PidNetONNXfloatSnapdragon® X Elite32.114 ms24 - 24 MBNPU
PidNetONNXfloatSnapdragon® 8 Gen 3 Mobile22.848 ms30 - 331 MBNPU
PidNetONNXfloatSnapdragon® 8 Gen 1 Mobile54.789 ms13 - 318 MBNPU
PidNetONNXfloatQualcomm® Dragonwing™ IQ-827540.42 ms24 - 51 MBNPU
PidNetONNXfloatQualcomm® Dragonwing™ QCS8550 (Proxy)32.832 ms24 - 46 MBNPU
PidNetONNXfloatQualcomm® QCS845054.789 ms13 - 318 MBNPU
PidNetONNXfloatQualcomm® Dragonwing™ IQ-907542.261 ms24 - 51 MBNPU
PidNetONNXfloatQualcomm® Dragonwing™ IQ-X718132.114 ms24 - 24 MBNPU
PidNetONNXfloatQualcomm® Dragonwing™ Q-875017.614 ms6 - 219 MBNPU
PidNetONNXfloatSnapdragon® 8 Elite Mobile17.614 ms6 - 219 MBNPU
PidNetONNXfloatSnapdragon® 8 Elite Gen 5 Mobile13.235 ms7 - 256 MBNPU
PidNetONNXw8a8Snapdragon® X2 Elite26.075 ms7 - 7 MBNPU
PidNetONNXw8a8Snapdragon® X Elite54.893 ms7 - 7 MBNPU
PidNetONNXw8a8Snapdragon® 8 Gen 3 Mobile38.835 ms7 - 270 MBNPU
PidNetONNXw8a8Snapdragon® 8 Gen 1 Mobile57.517 ms8 - 271 MBNPU
PidNetONNXw8a8Qualcomm® Dragonwing™ QCS6490157.864 ms6 - 9 MBNPU
PidNetONNXw8a8Qualcomm® Dragonwing™ IQ-827551.006 ms6 - 10 MBNPU
PidNetONNXw8a8Qualcomm® Dragonwing™ QCS8550 (Proxy)52.717 ms6 - 177 MBNPU
PidNetONNXw8a8Qualcomm® QCS845057.517 ms8 - 271 MBNPU
PidNetONNXw8a8Qualcomm® Dragonwing™ IQ-907554.064 ms6 - 9 MBNPU
PidNetONNXw8a8Qualcomm® Dragonwing™ IQ-X718154.893 ms7 - 7 MBNPU
PidNetONNXw8a8Qualcomm® Dragonwing™ Q-6690210.332 ms6 - 225 MBNPU
PidNetONNXw8a8Qualcomm® Dragonwing™ Q-779062.654 ms7 - 235 MBNPU
PidNetONNXw8a8Qualcomm® Dragonwing™ Q-875044.899 ms1 - 217 MBNPU
PidNetONNXw8a8Snapdragon® 8 Elite Mobile44.899 ms1 - 217 MBNPU
PidNetONNXw8a8Snapdragon® 8 Elite Gen 5 Mobile25.8 ms2 - 236 MBNPU
PidNetONNXw8a8Snapdragon® 7 Gen 4 Mobile62.654 ms7 - 235 MBNPU
PidNetQNN_DLCfloatSnapdragon® X2 Elite13.562 ms24 - 24 MBNPU
PidNetQNN_DLCfloatSnapdragon® X Elite38.875 ms24 - 24 MBNPU
PidNetQNN_DLCfloatSnapdragon® 8 Gen 3 Mobile25.468 ms16 - 323 MBNPU
PidNetQNN_DLCfloatSnapdragon® 8 Gen 1 Mobile75.957 ms3 - 308 MBNPU
PidNetQNN_DLCfloatQualcomm® Dragonwing™ IQ-827544.511 ms24 - 53 MBNPU
PidNetQNN_DLCfloatQualcomm® Dragonwing™ QCS8550 (Proxy)37.729 ms24 - 27 MBNPU
PidNetQNN_DLCfloatQualcomm® SA8775P46.605 ms24 - 247 MBNPU
PidNetQNN_DLCfloatQualcomm® SA8650P46.605 ms24 - 247 MBNPU
PidNetQNN_DLCfloatQualcomm® SA8255P46.605 ms24 - 247 MBNPU
PidNetQNN_DLCfloatQualcomm® QCS845075.957 ms3 - 308 MBNPU
PidNetQNN_DLCfloatQualcomm® Dragonwing™ IQ-907547.528 ms24 - 52 MBNPU
PidNetQNN_DLCfloatQualcomm® Dragonwing™ IQ-X718138.875 ms24 - 24 MBNPU
PidNetQNN_DLCfloatQualcomm® Dragonwing™ Q-875018.213 ms23 - 267 MBNPU
PidNetQNN_DLCfloatQualcomm® SA7255P117.751 ms24 - 246 MBNPU
PidNetQNN_DLCfloatQualcomm® SA8295P51.907 ms24 - 252 MBNPU
PidNetQNN_DLCfloatSnapdragon® 8 Elite Mobile18.213 ms23 - 267 MBNPU
PidNetQNN_DLCfloatSnapdragon® 8 Elite Gen 5 Mobile12.062 ms18 - 289 MBNPU
PidNetQNN_DLCw8a8Snapdragon® X2 Elite25.762 ms6 - 6 MBNPU
PidNetQNN_DLCw8a8Snapdragon® X Elite60.63 ms6 - 6 MBNPU
PidNetQNN_DLCw8a8Snapdragon® 8 Gen 3 Mobile43.111 ms6 - 267 MBNPU
PidNetQNN_DLCw8a8Snapdragon® 8 Gen 1 Mobile63.574 ms6 - 270 MBNPU
PidNetQNN_DLCw8a8Qualcomm® Dragonwing™ IQ-827556.547 ms6 - 15 MBNPU
PidNetQNN_DLCw8a8Qualcomm® Dragonwing™ QCS8550 (Proxy)57.901 ms6 - 8 MBNPU
PidNetQNN_DLCw8a8Qualcomm® SA8775P58.537 ms6 - 219 MBNPU
PidNetQNN_DLCw8a8Qualcomm® SA8650P58.537 ms6 - 219 MBNPU
PidNetQNN_DLCw8a8Qualcomm® SA8255P58.537 ms6 - 219 MBNPU
PidNetQNN_DLCw8a8Qualcomm® QCS845063.574 ms6 - 270 MBNPU
PidNetQNN_DLCw8a8Qualcomm® Dragonwing™ IQ-907559.064 ms6 - 14 MBNPU
PidNetQNN_DLCw8a8Qualcomm® Dragonwing™ IQ-X718160.63 ms6 - 6 MBNPU
PidNetQNN_DLCw8a8Qualcomm® Dragonwing™ Q-875043.833 ms6 - 241 MBNPU
PidNetQNN_DLCw8a8Qualcomm® SA7255P111.291 ms6 - 218 MBNPU
PidNetQNN_DLCw8a8Qualcomm® SA8295P66.453 ms6 - 223 MBNPU
PidNetQNN_DLCw8a8Snapdragon® 8 Elite Mobile43.833 ms6 - 241 MBNPU
PidNetQNN_DLCw8a8Snapdragon® 8 Elite Gen 5 Mobile24.54 ms6 - 272 MBNPU
PidNetTFLITEfloatSnapdragon® 8 Gen 3 Mobile24.955 ms2 - 322 MBNPU
PidNetTFLITEfloatSnapdragon® 8 Gen 1 Mobile74.253 ms0 - 326 MBNPU
PidNetTFLITEfloatQualcomm® Dragonwing™ IQ-827543.808 ms0 - 46 MBNPU
PidNetTFLITEfloatQualcomm® Dragonwing™ QCS8550 (Proxy)36.276 ms2 - 5 MBNPU
PidNetTFLITEfloatQualcomm® SA8775P45.283 ms3 - 233 MBNPU
PidNetTFLITEfloatQualcomm® SA8650P45.283 ms3 - 233 MBNPU
PidNetTFLITEfloatQualcomm® SA8255P45.283 ms3 - 233 MBNPU
PidNetTFLITEfloatQualcomm® QCS845074.253 ms0 - 326 MBNPU
PidNetTFLITEfloatQualcomm® Dragonwing™ IQ-907545.833 ms0 - 45 MBNPU
PidNetTFLITEfloatQualcomm® Dragonwing™ Q-875018.901 ms2 - 255 MBNPU
PidNetTFLITEfloatQualcomm® SA7255P116.628 ms0 - 228 MBNPU
PidNetTFLITEfloatQualcomm® SA8295P50.971 ms3 - 245 MBNPU
PidNetTFLITEfloatSnapdragon® 8 Elite Mobile18.901 ms2 - 255 MBNPU
PidNetTFLITEfloatSnapdragon® 8 Elite Gen 5 Mobile12.648 ms4 - 274 MBNPU
PidNetTFLITEw8a8Snapdragon® 8 Gen 3 Mobile38.123 ms1 - 262 MBNPU
PidNetTFLITEw8a8Snapdragon® 8 Gen 1 Mobile58.254 ms1 - 266 MBNPU
PidNetTFLITEw8a8Qualcomm® Dragonwing™ QCS6490200.632 ms2 - 72 MBNPU
PidNetTFLITEw8a8Qualcomm® Dragonwing™ IQ-827549.767 ms1 - 18 MBNPU
PidNetTFLITEw8a8Qualcomm® Dragonwing™ QCS8550 (Proxy)50.412 ms0 - 3 MBNPU
PidNetTFLITEw8a8Qualcomm® SA8775P51.017 ms1 - 214 MBNPU
PidNetTFLITEw8a8Qualcomm® SA8650P51.017 ms1 - 214 MBNPU
PidNetTFLITEw8a8Qualcomm® SA8255P51.017 ms1 - 214 MBNPU
PidNetTFLITEw8a8Qualcomm® QCS845058.254 ms1 - 266 MBNPU
PidNetTFLITEw8a8Qualcomm® Dragonwing™ IQ-907551.628 ms0 - 17 MBNPU
PidNetTFLITEw8a8Qualcomm® Dragonwing™ Q-6690243.409 ms3 - 233 MBNPU
PidNetTFLITEw8a8Qualcomm® Dragonwing™ Q-779066.153 ms1 - 219 MBNPU
PidNetTFLITEw8a8Qualcomm® Dragonwing™ Q-875070.041 ms1 - 233 MBNPU
PidNetTFLITEw8a8Qualcomm® SA7255P98.332 ms0 - 212 MBNPU
PidNetTFLITEw8a8Qualcomm® SA8295P58.069 ms1 - 217 MBNPU
PidNetTFLITEw8a8Snapdragon® 8 Elite Mobile70.041 ms1 - 233 MBNPU
PidNetTFLITEw8a8Snapdragon® 8 Elite Gen 5 Mobile22.173 ms0 - 266 MBNPU
PidNetTFLITEw8a8Snapdragon® 7 Gen 4 Mobile66.153 ms1 - 219 MBNPU

License

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

References

Community