Qualcomm® AI HubAI Hub

AG-Pose

Category‑level 6D object pose estimation.

AG‑Pose is a category‑level 6D object pose estimation model that predicts rotation, translation, and size of objects from RGB‑D input. It uses DINOv2‑S (ViT‑S/14) as the RGB backbone and PointNet++ for 3D point cloud processing, with an Instance‑Adaptive Keypoint Detector (IAKD) and Geometric‑Aware Feature Aggregator (GAFA) for robust pose estimation across 6 object categories.

Not supported

This model is currently not supported on any Mobile chipset.

To see performance metrics for this model on other chipsets, click the button below.

View for other chipsets

Technical Details

Input resolution:1 x 3 x 224 x 224
Model checkpoint:epoch_30.pt
Model size (float):~116 MB
Number of parameters:30.4M
Supported categories:bottle, bowl, camera, can, laptop, mug

Applicable Scenarios

  • Robotic manipulation
  • Augmented reality
  • Scene understanding

License

Model:MIT

Supported Mobile Devices

  • Samsung Galaxy S21
  • Samsung Galaxy S21 Ultra
  • Samsung Galaxy S22 5G
  • Samsung Galaxy S22 Ultra 5G
  • Samsung Galaxy S22+ 5G
  • Samsung Galaxy S23
  • Samsung Galaxy S23 Ultra
  • Samsung Galaxy S23+
  • Samsung Galaxy S24
  • Samsung Galaxy S24 Ultra
  • Samsung Galaxy S24+
  • Samsung Galaxy S25
  • Samsung Galaxy S25 Ultra
  • Samsung Galaxy S25+
  • Samsung Galaxy Tab S8
  • Xiaomi 12

Supported Mobile Chipsets

  • Snapdragon® 8 Elite For Galaxy Mobile
  • Snapdragon® 8 Gen 1 Mobile
  • Snapdragon® 8 Gen 2 Mobile
  • Snapdragon® 8 Gen 3 Mobile
  • Snapdragon® 888 Mobile

Looking for more? See models created by industry leaders.

Discover Model Makers