Back to the paper library

Object Pose Estimation · 2024 · WACV · CCF C

Robust Category-Level 3D Pose Estimation From Diffusion-Enhanced Synthetic Data

P3D-Diffusion fixes category-level 3D rotation labels with a renderer and augments appearance using Canny-edge-conditioned diffusion, DTD texture priors, and category-independent random backgrounds. CC3D adapts synthetic features to unlabeled real images through neural-mesh vertex correspondences. On PASCAL3D+, 10% real rotation labels yield 30° / 10° accuracy of 86.7 / 62.4 and MedErr 8.4°; 50% labels yield 90.7 / 71.4 / 6.9°. Translation / scale are not estimated, and diffusion-induced pose-label drift is not directly quantified.

Research area

Object Pose Estimation

Task level

Category-level

Object setting

Seen objects

Input modalities

single RGB object image at inferencecategory CAD models and rendered edge maps for synthetic trainingunlabeled real RGB images for CC3D adaptationoptional sparsely labeled real RGB images

Requirements

known object category and PASCAL3D+-style object cropcategory-level CAD models and cuboid neural meshgraphics renderer and edge-conditioned diffusion model for data generationunlabeled target-domain imagestest-time neural-mesh pose optimizationoptional 3D rotation labels for few-shot fine-tuning
Full paper metadata Authors, affiliations, research area, datasets, metrics, method tags, and reading status Expand
Authors
Jiahao Yang, Wufei Ma, Angtian Wang, Xiaoding Yuan, Alan Yuille, Adam Kortylewski
Affiliations
Peking University / Johns Hopkins University / University of Freiburg / Max Planck Institute for Informatics
Conference / journal
Conference · WACV · CCF C
Research area
Object Pose Estimation
Code availability
Code available
Reading status
Overview:Available Deep Dive:Available
Task setting
Category-level · Seen objects
Evaluation metrics
Rotation Error n° mcm

Original paper terminology:rotation accuracy below 30 degrees / rotation accuracy below 10 degrees / median angular error / OOD nuisance-wise rotation accuracy

Requirements
known object category and PASCAL3D+-style object cropcategory-level CAD models and cuboid neural meshgraphics renderer and edge-conditioned diffusion model for data generationunlabeled target-domain imagestest-time neural-mesh pose optimizationoptional 3D rotation labels for few-shot fine-tuning

Reading Notes

Overview / Deep Dive

Available src/paper-notes/p3d-diffusion-wacv2024/coarse.md

Paper overview notes

Quickly assess the paper's scope, problem setting, core method, experimental findings, and whether to read it in depth.

Paper Reading Notes (Coarse): Robust Category-Level 3D Pose Estimation From Diffusion-Enhanced Synthetic Data

1. Basic information

  • Short name: P3D-Diffusion + CC3D
  • Authors: Jiahao Yang, Wufei Ma, Angtian Wang, Xiaoding Yuan, Alan Yuille, Adam Kortylewski
  • Affiliations: Peking University, Johns Hopkins University, University of Freiburg, Max Planck Institute for Informatics
  • Year / venue: WACV 2024
  • Official page: CVF Open Access
  • Paper PDF: CVF PDF
  • Code: YangYY06/synthetic_3d
  • Reading mode: Overview

2. Positioning in one sentence

Precisely view-labeled CAD renders become edges that condition realistic diffusion appearances, with backgrounds randomized independently of categories. A 3D-aware Neural Mesh Model and CC3D then align unlabeled real-image vertex features to the synthetic domain, improving PASCAL3D+ / OOD-CV orientation with few real rotation labels.

This estimates 3D rotation (azimuth, elevation, in-plane rotation), without metric translation, size, or full 6D pose. It is category-level orientation estimation; Table 2 accuracies are neither ADD(-S) nor BOP 6D results.

2.1 Guide to the paper’s figures and tables

Figure / tableReading focus
Figure 1CAD renders, edge-guided diffusion, background randomization, and OOD evaluation
Figure 2Naive style-transfer color / texture collapse; DTD priors and prompts for diversity
Table 2PASCAL3D+ versus full supervision with 0%, 10%, 50% real rotation labels
Table 3OOD-CV shape, pose, texture, context, and weather shifts
P3D-Diffusion Figure 1: data generation and OOD-aware training overview
Original Figure 1: random $SO(3)$ CAD views render labels; Canny conditioning preserves pose structure during diffusion styling. Independently sampled backgrounds reduce category-context shortcuts; CC3D adapts neural meshes to real images. Source: CVF Open Access. Image copyright belongs to the paper's authors and publisher; reproduced here for academic analysis.

3. Problem setting

  • Task: category-level 3D orientation, allowing within-category geometry / appearance variation.
  • Training resources: category CAD, sampled rotations, Blender renders, DTD textures, 100 internet backgrounds, unlabeled real images; optionally 7 / 20 / 50 real labels per category, or 10% / 50% of full data.
  • Inference: one real object RGB image, using Neural Mesh Model / PASCAL3D+ detection and crop protocols.
  • Output: rotation matrices or azimuth / elevation / in-plane parameters.
  • Not output: translation, metric size, full 6D pose, or instance-CAD registration.
  • Objects: known train / test categories, mainly PASCAL3D+ / OOD-CV manufactured objects.

4. Core motivation

Real 3D labels are costly; CAD provides free labels with substantial sim-to-real gaps. The design separates:

  1. Controllable geometry labels: renderer rotations.
  2. Realistic / diverse appearance: edge-guided diffusion, DTD, random-color prompts.
  3. Cross-domain correspondence: CC3D vertex consistency.

Real data often encode car-road / boat-water shortcuts. Independently pairing 100 backgrounds with categories breaks correlations during generation rather than relying solely on model regularization.

5. Method overview

5.1 P3D-Diffusion data generation

For CAD CiC_i, rotation ξ∈SO(3)\xi\in SO(3), and background BjB_j:

  1. Blender renders Irender=R(Ci,ξ)I_{render}=R(C_i,\xi).
  2. Canny extracts edges EE.
  3. Pretrained ControlNet-style conditional diffusion Ψ(E)\Psi(E) produces realistic objects.
  4. Composite with BjB_j.
  5. Inherit renderer ξ\xi as rotation labels.

To avoid uniformly similar, textureless generations:

  • Random DTD textures produce 3D-consistent prior noise in edges.
  • Random color / category prompts, e.g. red car, green aeroplane.
  • 7,000 synthetic images per category.
  • Random selection among 100 category-independent internet backgrounds.
P3D-Diffusion Figure 2: generation-diversity design
Original Figure 2: naive transfer gives similar colors / weak texture; DTD prior noise and random-color prompts diversify appearance. Visual diversity does not quantitatively establish unchanged rotation labels for every generated image. Source: CVF Open Access. Image copyright belongs to the paper's authors and publisher; reproduced here for academic analysis.

5.2 Neural Mesh Model and CC3D

Category cuboid meshes store learnable features per vertex. Known training poses relate ResNet50 image features to vertices; inference optimizes pose by gradient descent to match rendered vertex and image features.

CC3D adaptation:

  1. Train neural meshes on synthetic images / exact rotations.
  2. Optimize real unlabeled poses for pseudo-labels.
  3. Cosine mesh-image similarity measures self-consistency confidence.
  4. Select 100 highest-confidence real images per category.
  5. Cross-domain vertex losses align real projected features with synthetic vertices.
  6. Joint contrastive, reconstruction, and domain-consistency optimization.

6. Experimental overview

6.1 Data and metrics

  • PASCAL3D+: 11,045 train, 10,812 validation, 12 categories; few-shot Table 1 uses only 6 vehicle categories.
  • OOD-CV: 10 PASCAL3D+ categories with single-factor shape, pose, texture, context, or weather shifts.
  • Metrics: geodesic rotation error; ACC π/6 below 30°, ACC π/18 below 10°; lower median angular MedErr is better.

6.2 Main PASCAL3D+ results

P3D-Diffusion Table 2: PASCAL3D+ rotation results
Original Table 2: CC3D without real labels gives 76.3 / 41.4 / 15.5; 10% real labels give 86.7 / 62.4 / 8.4, slightly exceeding supervised NeMo's strict accuracy 61.0; 50% gives 90.7 / 71.4 / 6.9. Columns are 30° accuracy, 10° accuracy, median angular error. Source: CVF Open Access. Image copyright belongs to the paper's authors and publisher; reproduced here for academic analysis.
TrainingACC 30°ACC 10°MedErr
NeMo full supervision86.161.08.8°
P3D-Diffusion + CC3D, no real pose labels76.341.415.5°
P3D-Diffusion + CC3D + 10% labels86.762.48.4°
P3D-Diffusion + CC3D + 50% labels90.771.46.9°

Matching full supervision with 10% and exceeding it with 50% holds within Table 2, with unequal total data: CC3D also has 7,000 synthetic images per category and unlabeled real images. This measures manual 3D label efficiency, not total data / compute efficiency.

6.3 OOD-CV

P3D-Diffusion Table 3: five OOD-CV shifts
Original Table 3: without real labels, CC3D mean 30° / 10° accuracy is 48.2 / 14.8, above same-data P3D-NeMo 42.6 / 12.7. With 10% labels it reaches 54.9 / 27.3; strict mean leads, but loose mean trails P3D-NeMo +10%'s 55.2. Not every OOD metric leads. Source: CVF Open Access. Image copyright belongs to the paper's authors and publisher; reproduced here for academic analysis.
  • No real labels: CC3D ACC 30° mean = 48.2, near supervised Res50 46.9 but below NeMo 51.6; ACC 10° mean = 14.8, above Res50 14.3, below NeMo 24.5.
  • With 10% labels: 54.9 / 27.3; loose mean trails P3D-NeMo 55.2, strict mean improves.
  • Texture / weather shifts remain weak; the authors note insufficient synthetic-texture realism.

7. Strengths and limitations

Strengths

  • Separates realism, label preservation, shortcuts, and cross-domain correspondence.
  • Evaluates five OOD factors beyond in-domain PASCAL3D+.
  • 10% / 50% experiments suggest substantial manual-label savings.
  • Styling and unsupervised adaptation help independently; removing both hurts most.
  • Public code supports generation / CC3D reproduction.

Limitations / risks

  • Rotation alone cannot directly serve metric-translation grasping / assembly.
  • Edges must preserve geometry to inherit poses, but post-generation label drift is not measured.
  • Only 100 internet backgrounds, with source, copyright, resolution, and deduplication mainly in supplements; long-tail coverage is limited.
  • Analysis-by-synthesis optimization, not one-pass inference; no systematic runtime / initialization analysis.
  • Top-100 pseudo-label selection may reinforce initial biases and underrepresent rare views.
  • Manufactured PASCAL3D+ / OOD-CV objects do not cover indoor occlusion, transparent / reflective objects, instance CAD registration, or BOP.

8. Research takeaways

  • Generation: fix geometric labels first, generate appearance second; conditioning must preserve pose-sensitive structure.
  • Generalization: break category-background correlations as well as randomizing texture.
  • Design: render-and-compare consistency can filter pseudo-labels beyond classifier confidence with 3D templates.
  • Baseline: category orientation, few-shot pose, and sim-to-real / OOD studies; not direct ADD(-S) / BOP instance-6D comparisons.
  • Deep-reading value: diffusion training data, weak rotation supervision, and OOD generalization.

9. Coarse-reading conclusion

Diffusion produces data with precise renderer labels, rather than directly predicting pose; neural meshes constrain synthetic-real correspondence. Its label efficiency / OOD evidence is useful, provided it remains category-level single-image orientation, without translation or guarantees of zero label drift.

Related Papers

Related papers