SchemaPose: RGB-Based Category-Level Pose Estimation with Parametric Category Schema

Shanghai Jiao Tong University

* Equal contribution† Corresponding author

NeurIPS 2026

Abstract

RGB-based category-level 9D pose estimation remains highly challenging because a monocular system must generalize across unseen object instances while jointly inferring 3D translation, 3D rotation, and 3D size. Unlike instance-level pose estimation, the category-level setting requires the network to learn both what is shared across objects within a category and how individual instances vary in structure. This challenge is amplified in the RGB-only setting, where pose cues, category-level commonality, and instance-specific variation are entangled in appearance without direct geometric constraints, making it difficult for existing frameworks to learn a representation that is truly aligned with the category-level estimation objective. To address this problem, we propose SchemaPose, a schema-guided framework for RGB-based category-level pose estimation. SchemaPose equips the estimation pipeline with an explicit parametric schema for modeling category-level regularities together with structured instance variation, and uses the inferred schema variables to guide pose-related prediction in an end-to-end query-based framework. The same formulation also supports template instantiation, synthetic annotation, and pose learning within a unified framework. Experiments on synthetic and real data show that SchemaPose achieves state-of-the-art performance among RGB-based methods for category-level 9D pose estimation.

Parametric Category Schema

Figure 1. Overview of the parametric category schema, compact representation, and data generation pipeline. (a) Parametric Category Schema. For category c, the schema comprises discrete part choices dc and continuous geometric parameters xc, which define a valid object structure via semantic part templates. (b) Compact Representation for Learning. A category-specific compression function Ψc maps the full schema variable zc=(dc,xc) to a compact vector gc, preserving the dominant pose-relevant structural factors for network training. (c) Schema-Driven Data Generation. The schema variable zc is analytically instantiated into an object model Mc via Φc. These models are diversified through parameter perturbation and rendered within synthetic scenes to generate training images with aligned ground-truth pose labels.
Figure 2. Relationship between object schema and pose. Objects are parameterized through schema variables, which provide the structural basis for expressing and interpreting 9D pose.

Method

Figure 3. Overview of SchemaPose and Dual-Stage Schema Guided Mechanism. (a) SchemaPose follows an end-to-end query-based pipeline. Given an RGB image, the backbone and Transformer encoder produce encoder memory Qenc, from which top-K proposals are selected to initialize decoder queries. The Transformer decoder refines these queries and produces decoder features Qdec. (b) The Schema-Guided Pose Head (SGPH) follows the same design at the proposal and decoder stages. Given an object feature q, SGPH first predicts the object category c and bounding box b, then infers a unified schema vector g. A category-specific mask mc selects the effective schema representation g̃, which is fed back as structural guidance for pose-related prediction. During training, detection, schema, and pose predictions are supervised at both stages, while inference uses the final pose output (ox,oy), z, R, s from the last decoder layer.

Results

Scroll horizontally to see all metrics

Table 1. Comparison with state-of-the-art RGB-based methods on CAMERA25 and REAL275. The best result in each column is shown in bold.
MethodCAMERA25REAL275
IoU50IoU7510cm10°10°10cmIoU50IoU7510cm10°10°10cm
MSOS32.45.129.760.819.223.43.039.529.29.6
OLD-Net32.15.430.174.023.425.41.938.937.09.8
DMSR34.66.532.381.427.428.36.137.359.523.6
LaPose—————17.52.644.4—30.5
GIVEPose—————20.1—45.9—34.2
MonoDiff9D35.26.733.680.128.231.56.341.056.325.7
DA-Pose41.46.140.560.824.628.13.645.827.513.4
Fischer et al.—————17.52.6——29.4
LanCOPE—————35.46.646.558.730.0
YOPO46.611.843.188.738.771.616.477.869.652.8
SchemaPose (Ours)52.413.047.989.043.186.324.691.175.468.4
Figure 4. Qualitative comparison with YOPO on REAL275. Green boxes indicate ground truth and red boxes indicate predictions. Light-blue arrows mark instances where YOPO fails while SchemaPose produces more accurate pose estimates.

Video Results

Additional Scenes

2 videos
Scene 10:28
Scene 20:17

REAL275

6 videos

Ground truth (left) and predictions (right).

Scene 10:35
Scene 20:49
Scene 30:49
Scene 40:44
Scene 50:45
Scene 60:54

Figure