Back to the paper library

Object Pose Estimation · 2024 · WACV · CCF C

Learning Better Keypoints for Multi-Object 6DoF Pose Estimation

KeyGNet replaces fixed FPS / BBox keypoints with offline target design. EdgeConv learns shared 3D keypoints from visible colored object surfaces within a dataset. Wasserstein loss aligns voting distributions across keypoints, and dispersion loss prevents geometric collapse. Replacing keypoints in PVNet, PVN3D, and RCVPose raises MIMO mean AR across seven BOP core datasets by 11.3, 12.0, and 10.0 points; PVN3D LMO ADD(S) rises from 53.9 to 70.3. Benefits center on closing the SISO→MIMO gap, but ground-truth poses, intrinsics, known geometry, visible surfaces, and separate training per dataset / voting representation are required.

Research area

Object Pose Estimation

Task level

Instance-level

Object setting

Seen objects

Input modalities

colored visible object-surface points for KeyGNet trainingRGB input for PVNetRGB-D / point-cloud input for PVN3D and RCVPose

Requirements

known object geometry during keypoint optimizationtraining images with ground-truth object poses and camera intrinsicsvisible object-surface segmentsthe detector / mask / depth dependencies of the selected downstream pose estimatorseparate KeyGNet optimization for each dataset and voting representation
Full paper metadata Authors, affiliations, research area, datasets, metrics, method tags, and reading status Expand
Authors
Yangzheng Wu, Michael Greenspan
Affiliations
Queen's University
Conference / journal
Conference · WACV · CCF C
Research area
Object Pose Estimation
Code availability
Code available
Reading status
Overview:Available Deep Dive:Available
Task setting
Instance-level · Seen objects
Evaluation metrics
BOP AR ADD ADD-S ADD / ADD-S AUC ADD(S) / ADD(-S)

Original paper terminology:BOP Average Recall / VSD recall / MSSD recall / MSPD recall / ADD(S) / ADD(S) AUC / SISO-to-MIMO performance gap / training time

Requirements
known object geometry during keypoint optimizationtraining images with ground-truth object poses and camera intrinsicsvisible object-surface segmentsthe detector / mask / depth dependencies of the selected downstream pose estimatorseparate KeyGNet optimization for each dataset and voting representation

Reading Notes

Overview / Deep Dive

Available src/paper-notes/keygnet-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): Learning Better Keypoints for Multi-Object 6DoF Pose Estimation

1. Basic information

  • Title: Learning Better Keypoints for Multi-Object 6DoF Pose Estimation
  • Short name: KeyGNet
  • Authors: Yangzheng Wu, Michael Greenspan
  • Affiliation: Queen’s University, RCV Lab / Ingenuity Labs
  • Year / venue: WACV 2024
  • Task: learn predefined keypoint layouts for known rigid objects in multi-object 6DoF pose estimation
  • Paper: CVF Open Access · PDF · arXiv
  • Code: aaronWool/keygnet
  • Reading mode: Overview

2. Positioning in one sentence

KeyGNet replaces fixed FPS surface points or bounding-box corners with shared keypoints learned by a graph network from training objects’ visible colored surfaces. It makes keypoint vote distributions similar while keeping positions dispersed. Replacing heuristic keypoints in PVNet, PVN3D, and RCVPose substantially reduces the SISO→MIMO accuracy loss in multi-object training.

The paper revisits a design choice often treated as constant: which 3D keypoints should the downstream network regress? It does not introduce an entirely new pose pipeline.

2.1 Guide to the paper’s figures and tables

Figure / tableReading focus
Figure 3KeyGNet inputs, graph-network outputs, and two optimization objectives: the central method diagram.
Table 1BOP AR for three voting methods, seven BOP core datasets, and SISO / MIMO training: test gains across methods and datasets.
Table 2LMO ADD(S), YCB-V ADD(S) AUC, and training time: examine accuracy and convergence gains together.
Table 5KL, JS, cross-entropy, and Wasserstein distribution-similarity losses: test the core loss choice.
KeyGNet Figure 3: keypoint graph-network architecture
Original Figure 3: visible colored surface points from training objects enter a graph network that outputs keypoints in normalized object coordinates. The Wasserstein term aligns their vote distributions, while dispersion prevents local collapse. Source: CVF Open Access. Image copyright belongs to the paper's authors and publisher; reproduced here for academic analysis.

3. Problem setting

3.1 Task Boundaries

  • Object setting: instance-level, seen-object; dataset objects are known during training.
  • KeyGNet training input: unoccluded visible colored surface points recovered from training images, intrinsics, GT object poses, and object geometry.
  • KeyGNet output: 3D keypoints in normalized object coordinates; multiple objects in a dataset share keypoint indices / layouts.
  • Downstream inputs: estimator-dependent. PVNet primarily uses RGB; PVN3D and RCVPose use RGB-D / point-cloud information.
  • Final output: the downstream method still estimates object 3D rotation and 3D translation; KeyGNet does not directly output pose.
  • Dependencies: training needs object surface geometry, GT poses, camera calibration, and visible-surface segmentation. Inference retains the downstream estimator’s original inputs, detection / segmentation, and geometric-solver requirements.

3.2 SISO and MIMO

  • SISO: Single Instance of a Single Object; train separate network parameters for each object.
  • MIMO: Multiple Instances of Multiple Objects; multiple objects share a network, closer to multi-object deployment.
  • Switching fixed FPS / BBox keypoints from SISO to MIMO increases vote-regression distribution heterogeneity; the paper observes substantial accuracy degradation.
  • KeyGNet aims to balance keypoint target distributions for the shared regressor and narrow this gap.

4. Core motivation

Traditional keypoint-based 6DoF methods typically use two fixed rules:

  1. FPS: select mutually distant surface points;
  2. BBox: select 3D bounding-box corners.

Both prioritize geometric dispersion without considering downstream regression-target distributions. PVNet regresses pixel-to-keypoint directions, PVN3D 3D offsets, and RCVPose radial distances. Large differences in target ranges create an optimization problem for a shared regressor analogous to class imbalance.

Using LINEMOD driller, the authors compare radial-vote histograms for three keypoints: smaller Wasserstein distances indicate more similar, jointly learnable distributions. This motivates two simultaneous objectives:

  • Distribution similarity: make keypoint vote distributions as similar as possible;
  • Geometric dispersion: keep keypoints sufficiently far apart to avoid degenerate pose geometry.

5. Method overview

5.1 What KeyGNet Does

KeyGNet uses EdgeConv / Dynamic Graph CNN. Training combines visible surface-point sets from all objects in a dataset and outputs NKN_K normalized 3D keypoints:

K∗={kj∗}j=1NK.K^*=\{k_j^*\}_{j=1}^{N_K}.

These replace the downstream network’s FPS or BBox keypoints. Because voting representations induce different distributions, KeyGNet is trained separately for PVNet, PVN3D, and RCVPose; one keypoint set does not transfer unconditionally to every estimator.

5.2 Two Losses

The total loss is:

L=αLwass+βLdis.\mathcal L=\alpha \mathcal L_{wass}+\beta \mathcal L_{dis}.
  • Lwass\mathcal L_{wass}: inspired by WGAN-GP, reduce Wasserstein distances between keypoint vote distributions with a gradient penalty.
  • Ldis\mathcal L_{dis}: penalize overly close keypoints through an exponential function of pairwise distance to prevent local clustering.
  • The first training half uses α=0.7,β=0.3\alpha=0.7,\beta=0.3, emphasizing vote distributions; after epoch 50, weights are swapped to emphasize geometric dispersion.

5.3 Training Configuration

  • batch size: 32;
  • Optimizer: SGD;
  • Initial learning rate: 10−310^{-3};
  • Multiply by 0.1 every 50 epochs;
  • Wasserstein gradient penalty: λ=10\lambda=10;
  • Dispersion hyperparameter: γ=0.5\gamma=0.5;
  • Parallel training on 6 RTX 6000 GPUs;
  • PVNet / PVN3D ultimately use 8 keypoints; RCVPose uses 3.

6. Experimental overview

6.1 Datasets and Metrics

The paper covers the seven BOP core datasets of its time: LMO, YCB-V, T-LESS, TUD-L, IC-BIN, ITODD, and HB. Main measures include:

  • BOP AR, aggregating VSD, MSSD, and MSPD;
  • LMO ADD(S);
  • YCB-V ADD(S) AUC;
  • Performance changes from SISO to MIMO;
  • Downstream pose-network training time.
KeyGNet Table 1: Average Recall on seven BOP datasets
Original Table 1: replacing FPS / BBox with KeyGNet (KGN) improves AR across three downstream networks, seven BOP core datasets, and SISO / MIMO settings. The largest individual gain is +21.5 points for PVN3D-MIMO on ITODD; mean MIMO gains across seven datasets are +11.3 for PVNet, +12.0 for PVN3D, and +10.0 for RCVPose. Source: CVF Open Access. Image copyright belongs to the paper's authors and publisher; reproduced here for academic analysis.

6.2 Main results

SettingHeuristic keypointsKeyGNetChange
PVN3D, LMO, MIMO, ADD(S)53.970.3+16.4
PVNet, YCB-V, MIMO, ADD(S) AUC65.178.1+13.0
PVNet, mean MIMO AR over seven BOP datasets61.973.2+11.3
PVN3D, mean MIMO AR over seven BOP datasets63.975.9+12.0
RCVPose, mean MIMO AR over seven BOP datasets70.280.2+10.0

The abstract’s +1% to +21.5% refers to absolute AR percentage-point differences across Table 1 datasets / methods / modes, not relative error reductions.

KeyGNet Table 2: LMO and YCB-V accuracy and training time
Original Table 2: KeyGNet improves accuracy and shortens downstream training under the paper's configuration. In MIMO, PVNet / PVN3D / RCVPose decrease from 22 / 23 / 22 hours to 12 / 12 / 11 hours. This compares convergence in a specific six-GPU environment, not a universal wall-clock guarantee. Source: CVF Open Access. Image copyright belongs to the paper's authors and publisher; reproduced here for academic analysis.

6.3 SISO→MIMO Gap

  • Mean LMO ADD(S) gap: PVNet shrinks from -9.1 to -0.1, PVN3D from -9.3 to -0.2, and RCVPose changes from -6.3 to +0.1.
  • Mean YCB-V ADD(S) AUC gaps shrink from -6.3 / -6.7 / -3.4 to -1.0 / -1.2 / -1.0.

This matters more than SISO gains alone: KeyGNet brings a shared multi-object regressor closer to separately trained per-object performance.

7. Strengths and limitations

Strengths

  • Identifies a performance variable hidden by fixed rules: keypoint positions can themselves be learned.
  • Consistently positive gains across three voting representations, seven BOP datasets, and both SISO / MIMO modes provide broad evidence.
  • Replaces 3D keypoint supervision without changing downstream backbones, making it a modular data / geometry design component.
  • Reports accuracy, training time, keypoint counts, losses, and initialization, providing a relatively complete evidence chain.

Limitations / risks

  • KeyGNet inputs are recovered using GT poses, intrinsics, and object geometry; training-data preparation is costly.
  • Keypoints must be reoptimized for each dataset and voting representation; generalization as a universal keypoint generator is untested.
  • Independent robustness tests for detection errors, noisy masks, missing depth, reflective / transparent objects, and real deployment domain shifts are insufficient.
  • Comparisons retrain complete downstream networks, so layout, optimization difficulty, and early stopping may all contribute; these factors are not fully isolated.
  • Training uses 6 RTX 6000 GPUs, without complete KeyGNet training-time, memory, or random-seed variance reports.

8. Research takeaways

  • Transferable idea: treat fixed pose-pipeline geometry—keypoints, template viewpoints, sample locations, correspondence weights—as learnable data-design variables.
  • Baseline suitability: useful for keypoint voting, shared multi-object networks, or SISO→MIMO scaling; not a direct baseline for methods without predefined keypoints.
  • Most useful reproduction: fix all pose-backbone training conditions, vary only FPS / BBox / KeyGNet, and report multiple seeds, convergence curves, and final accuracy.
  • Worth deep reading: yes; the value lies in problem formulation and cross-method validation beyond one WACV benchmark gain.
  • Further questions: can keypoints be conditioned on object instances, directly optimize symmetry-aware pose uncertainty, or jointly model foundation features, visibility, and occlusion risk?

9. Coarse Reading Conclusion

KeyGNet shows that keypoint-based pose accuracy depends both on how keypoints are predicted and on which keypoints the network is asked to predict. Extensive experiments support this within the studied known-object, offline-training, voting-based pipelines. It is not yet a keypoint generator for unknown objects, CAD-free use, or direct cross-dataset generalization.

Related Papers

Related papers