arXiv:2603.25351cs.CVcs.AI2026-03

比较五种旋转估计方法,发现概率模型更稳定,分类法精度高但训练难。

Image Rotation Angle Estimation: Comparing Circular-Aware Methods

  • 用圆形损失、向量回归等五种方法处理角度循环特性
  • 最优模型在DRC-D数据集上误差仅1.23°,COCO数据集达2.84°
  • 适合需要高鲁棒性的图像预处理场景,如文档分析

自动图像旋转估计是众多视觉流水线中的关键预处理步骤。由于角度具有循环拓扑结构,导致边界不连续,标准回归方法难以胜任。本文系统比较了五种针对循环特性的方法:带圆形损失的直接角度回归、基于角度分箱的分类、单位向量回归、相位移编码器以及圆形高斯分布。通过使用ImageNet预训练模型进行迁移学习,我们对十六种现代架构进行了评估,调整其输出头以实现旋转预测。结果表明,概率方法(尤其是圆形高斯分布)在不同架构间表现最稳健;而分类方法在匹配良好的骨干网络上达到最高精度,但在其他架构上存在训练不稳定性。最佳配置(分类+EfficientViT-B3)在DRC-D数据集上平均绝对误差(MAE)为1.23°(五次独立运行均值),圆形高斯分布配合MambaOut Base同样达到1.24°,且跨架构鲁棒性更强。在COCO 2014上,最佳配置达到3.71° MAE,显著优于先前工作;在更大规模的COCO 2017上进一步提升至2.84°。

原文摘要 · Abstract (English)

Automatic image rotation estimation is a key preprocessing step in many vision pipelines. This task is challenging because angles have circular topology, creating boundary discontinuities that hinder standard regression methods. We present a comprehensive study of five circular-aware methods for global orientation estimation: direct angle regression with circular loss, classification via angular binning, unit-vector regression, phase-shifting coder, and circular Gaussian distribution. Using transfer learning from ImageNet-pretrained models, we systematically evaluate these methods across sixteen modern architectures by adapting their output heads for rotation-specific predictions. Our results show that probabilistic methods, particularly the circular Gaussian distribution, are the most robust across architectures, while classification achieves the best accuracy on well-matched backbones but suffers training instabilities on others. The best configuration (classification with EfficientViT-B3) achieves a mean absolute error (MAE) of 1.23° (mean across five independent runs) on the DRC-D dataset, while the circular Gaussian distribution with MambaOut Base achieves a virtually identical 1.24° with greater robustness across backbones. Training and evaluating our top-performing method-architecture combinations on COCO 2014, the best configuration reaches 3.71° MAE, improving substantially over prior work, with further improvement to 2.84° on the larger COCO 2017 dataset.

图像旋转角度估计概率建模迁移学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。