Skip to content

Latest commit

 

History

History
61 lines (38 loc) · 2.65 KB

README.md

File metadata and controls

61 lines (38 loc) · 2.65 KB

Qualcomm® AI Hub Models

RTMPose_Body2d: Human pose estimation

RTMPose is a machine learning model that detects human pose and returns a location and confidence for each of 133 joints.

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

Example & Usage

Install the package via pip:

pip install "qai-hub-models[rtmpose-body2d]" torch==2.4.1 -f https://download.openmmlab.com/mmcv/dist/cpu/torch2.4/index.html -f https://qaihub-public-python-wheels.s3.us-west-2.amazonaws.com/index.html

Once installed, run the following simple CLI demo:

python -m qai_hub_models.models.rtmpose_body2d.demo

More 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.

Export for on-device deployment

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.rtmpose_body2d.export

Additional options are documented with the --help option. Note that the above script requires access to Deployment instructions for Qualcomm® AI Hub.

License

  • The license for the original implementation of RTMPose_Body2d can be found here.
  • The license for the compiled assets for on-device deployment can be found here

References

Community