Diffusion-based models enable monocular geometry estimation, yet their pixel-space precision is limited by a shared, under-studied error source: VAE reconstruction degradation. The 8x spatial compression in the VAE encoder-decoder degrades surface normals at object boundaries; even encoding and decoding ground-truth normals introduces 1.3--8.5° of mean angular error (MAE), with edge MAE reaching 2.8x the global MAE. We present TransNormal-2, a FLUX.2-based rectified-flow framework with single-step deterministic inference that addresses this degradation on both sides of the VAE decoder: in how latent predictions are supervised during training, and in how decoded normals are corrected at inference. First, geometry-aware pixel-space losses, including inverse rendering self-consistency, von~Mises-Fisher angular loss, and wavelet edge-aware regularization, complement latent MSE by enforcing spherical normal geometry and diffuse image-formation cues after VAE decoding. Second, a lightweight Geometric Refinement Module (GRM) applies an RGB-guided residual correction to reduce boundary-localized decoding errors without freely rewriting the coarse prediction. On general-scene benchmarks, TransNormal-2 matches or exceeds MoGe-2 on all eight reported metrics while using only 1.4% as many task-specific normal annotations. The gains are clearest for transparent objects, reducing MAE by 4.2° on ClearGrasp and 3.1° on ClearPose over the strongest prior baselines. Code will be released at https://longxiang-ai.github.io/TransNormal-2.
TransNormal-2:基于几何约束的修正流模型实现精确法线估计
AI 导读
TransNormal-2 提出一种基于 FLUX.2 的修正流框架,通过单步确定性推理解决 VAE 重建退化导致的法线估计精度问题。该方法在八个通用场景基准上达到或超越 MoGe-2,仅需其 1.4% 的任务特定法线标注;在透明物体数据集 ClearGrasp 和 ClearPose 上 MAE 分别降低 4.2° 和 3.1°。代码将开源。
HuggingFace Daily Papers(社区热门论文)
35
AI 编辑部评分,满分 100TransNormal-2:基于几何约束的修正流模型实现精确法线估计
TransNormal-2 提出一种基于 FLUX.2 的修正流框架,通过单步确定性推理解决 VAE 重建退化导致的法线估计精度问题。该方法在八个通用场景基准上达到或超越 MoGe-2,仅需其 1.4% 的任务特定法线标注;在透明物体数据集 ClearGrasp 和 ClearPose 上 MAE 分别降低 4.2° 和 3.1°。代码将开源。
原文 · 保持原样,未翻译原文 · 未翻译
来源:HuggingFace Daily Papers(社区热门论文)· arxiv.org