Hugging Face's logo

qualcomm
/
PSPNet

Image Segmentation
PyTorch
bu_auto
android
Model card Files
xet
 Community

PSPNet: Optimized for Qualcomm Devices

PSPNet (Pyramid Scene Parsing Network) is a semantic segmentation model that captures global context information by applying pyramid pooling modules. It is designed to improve scene understanding by aggregating contextual features at multiple scales.

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.45, ONNX Runtime 1.27.1 Download
QNN_DLC float Universal QAIRT 2.45 Download
TFLITE float Universal QAIRT 2.45 Download

For more device-specific assets and performance metrics, visit PSPNet 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 PSPNet on GitHub for usage instructions.

Model Details

Model Type: Model_use_case.semantic_segmentation

Model Stats:

  • Input resolution: 1x3x473x473
  • Model checkpoint: pspnet101_ade20k.pth
  • Model size (float): 251 MB
  • Number of parameters: 65.7M

Performance Summary

Model Runtime Precision Chipset Inference Time (ms) Peak Memory Range (MB) Primary Compute Unit
PSPNet ONNX float Snapdragon® X2 Elite 833.922 ms 532 - 532 MB NPU
PSPNet ONNX float Snapdragon® X Elite 1338.221 ms 267 - 267 MB NPU
PSPNet ONNX float Snapdragon® 8 Gen 3 Mobile 956.163 ms 207 - 2054 MB NPU
PSPNet ONNX float Snapdragon® 8 Gen 1 Mobile 2237.393 ms 33 - 887 MB NPU
PSPNet ONNX float Qualcomm® Dragonwing™ IQ-8275 1389.513 ms 117 - 123 MB NPU
PSPNet ONNX float Qualcomm® Dragonwing™ QCS8550 (Proxy) 1166.652 ms 0 - 160 MB NPU
PSPNet ONNX float Qualcomm® QCS8450 2237.393 ms 33 - 887 MB NPU
PSPNet ONNX float Qualcomm® Dragonwing™ IQ-9075 1684.754 ms 114 - 120 MB NPU
PSPNet ONNX float Qualcomm® Dragonwing™ IQ-X7181 1338.221 ms 267 - 267 MB NPU
PSPNet ONNX float Qualcomm® Dragonwing™ Q-8750 648.933 ms 170 - 1644 MB NPU
PSPNet ONNX float Snapdragon® 8 Elite Mobile 648.933 ms 170 - 1644 MB NPU
PSPNet ONNX float Snapdragon® 8 Elite Gen 5 Mobile 717.737 ms 117 - 1713 MB NPU
PSPNet QNN_DLC float Snapdragon® X2 Elite 2505.903 ms 3 - 3 MB NPU
PSPNet QNN_DLC float Snapdragon® X Elite 2540.633 ms 3 - 3 MB NPU
PSPNet QNN_DLC float Snapdragon® 8 Gen 3 Mobile 1847.35 ms 0 - 1654 MB NPU
PSPNet QNN_DLC float Snapdragon® 8 Gen 1 Mobile 1609.246 ms 0 - 852 MB NPU
PSPNet QNN_DLC float Qualcomm® Dragonwing™ IQ-8275 2531.233 ms 3 - 136 MB NPU
PSPNet QNN_DLC float Qualcomm® Dragonwing™ IQ-8275 5289.358 ms 0 - 1305 MB NPU
PSPNet QNN_DLC float Qualcomm® Dragonwing™ QCS8550 (Proxy) 2489.852 ms 3 - 5 MB NPU
PSPNet QNN_DLC float Qualcomm® SA8775P 2605.883 ms 1 - 1307 MB NPU
PSPNet QNN_DLC float Qualcomm® SA8650P 2605.883 ms 1 - 1307 MB NPU
PSPNet QNN_DLC float Qualcomm® SA8255P 2605.883 ms 1 - 1307 MB NPU
PSPNet QNN_DLC float Qualcomm® QCS8450 1609.246 ms 0 - 852 MB NPU
PSPNet QNN_DLC float Qualcomm® Dragonwing™ IQ-9075 2598.163 ms 3 - 135 MB NPU
PSPNet QNN_DLC float Qualcomm® Dragonwing™ IQ-X7181 2540.633 ms 3 - 3 MB NPU
PSPNet QNN_DLC float Qualcomm® Dragonwing™ Q-8750 2168.941 ms 0 - 1310 MB NPU
PSPNet QNN_DLC float Qualcomm® SA7255P 5289.358 ms 0 - 1305 MB NPU
PSPNet QNN_DLC float Qualcomm® SA8295P 1361.805 ms 3 - 647 MB NPU
PSPNet QNN_DLC float Snapdragon® 8 Elite Mobile 2168.941 ms 0 - 1310 MB NPU
PSPNet QNN_DLC float Snapdragon® 8 Elite Gen 5 Mobile 2341.485 ms 0 - 1363 MB NPU
PSPNet TFLITE float Snapdragon® 8 Gen 3 Mobile 2105.151 ms 42 - 1747 MB NPU
PSPNet TFLITE float Snapdragon® 8 Gen 1 Mobile 1861.665 ms 129 - 1060 MB NPU
PSPNet TFLITE float Qualcomm® Dragonwing™ IQ-8275 2879.024 ms 3 - 278 MB NPU
PSPNet TFLITE float Qualcomm® Dragonwing™ IQ-8275 5956.898 ms 130 - 1528 MB NPU
PSPNet TFLITE float Qualcomm® Dragonwing™ QCS8550 (Proxy) 2841.025 ms 0 - 4 MB NPU
PSPNet TFLITE float Qualcomm® SA8775P 2954.21 ms 123 - 1521 MB NPU
PSPNet TFLITE float Qualcomm® SA8650P 2954.21 ms 123 - 1521 MB NPU
PSPNet TFLITE float Qualcomm® SA8255P 2954.21 ms 123 - 1521 MB NPU
PSPNet TFLITE float Qualcomm® QCS8450 1861.665 ms 129 - 1060 MB NPU
PSPNet TFLITE float Qualcomm® Dragonwing™ IQ-9075 2922.32 ms 16 - 291 MB NPU
PSPNet TFLITE float Qualcomm® Dragonwing™ Q-8750 2165.438 ms 1 - 1409 MB NPU
PSPNet TFLITE float Qualcomm® SA7255P 5956.898 ms 130 - 1528 MB NPU
PSPNet TFLITE float Qualcomm® SA8295P 1421.541 ms 71 - 780 MB NPU
PSPNet TFLITE float Snapdragon® 8 Elite Mobile 2165.438 ms 1 - 1409 MB NPU
PSPNet TFLITE float Snapdragon® 8 Elite Gen 5 Mobile 2329.54 ms 5 - 1456 MB NPU

License

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

Community

Downloads last month

-

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

Model tree for qualcomm/PSPNet

Finetunes
1 model