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 IoT 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 IoT Devices

  • Dragonwing Q-8750

Supported IoT Chipsets

  • Qualcomm® Dragonwing™ Q-8750

Looking for more? See models created by industry leaders.

Discover Model Makers