EfficientViT is a machine learning model that can classify images from the Imagenet dataset. It can also be used as a backbone in building more complex models for specific use cases.
This is based on the implementation of EfficientViT-l2-cls found here. This repository contains scripts for optimized on-device export suitable to run on Qualcomm® devices. More details on model performance accross various devices, can be found here.
Sign up to start using Qualcomm AI Hub and run these models on a hosted Qualcomm® device.
Install the package via pip:
pip install "qai_hub_models[efficientvit_l2_cls]"Once installed, run the following simple CLI demo:
python -m qai_hub_models.models.efficientvit_l2_cls.demoMore details on the CLI tool can be found with the --help option. See
demo.py for sample usage of the model including pre/post processing
scripts. Please refer to our general instructions on using
models for more usage instructions.
This repository contains export scripts that produce a model optimized for on-device deployment. This can be run as follows:
python -m qai_hub_models.models.efficientvit_l2_cls.exportAdditional options are documented with the --help option. Note that the above
script requires access to Deployment instructions for Qualcomm® AI Hub.
- The license for the original implementation of EfficientViT-l2-cls can be found here.
- The license for the compiled assets for on-device deployment can be found here
- EfficientViT: Multi-Scale Linear Attention for High-Resolution Dense Prediction
- Source Model Implementation
- Join our AI Hub Slack community to collaborate, post questions and learn more about on-device AI.
- For questions or feedback please reach out to us.