Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Orient Anything: Learning Robust Object Orientation Estimation from Rendering 3D Models

About

Orientation is a key attribute of objects, crucial for understanding their spatial pose and arrangement in images. However, practical solutions for accurate orientation estimation from a single image remain underexplored. In this work, we introduce Orient Anything, the first expert and foundational model designed to estimate object orientation in a single- and free-view image. Due to the scarcity of labeled data, we propose extracting knowledge from the 3D world. By developing a pipeline to annotate the front face of 3D objects and render images from random views, we collect 2M images with precise orientation annotations. To fully leverage the dataset, we design a robust training objective that models the 3D orientation as probability distributions of three angles and predicts the object orientation by fitting these distributions. Besides, we employ several strategies to improve synthetic-to-real transfer. Our model achieves state-of-the-art orientation estimation accuracy in both rendered and real images and exhibits impressive zero-shot ability in various scenarios. More importantly, our model enhances many applications, such as comprehension and generation of complex spatial concepts and 3D object pose adjustment.

Zehan Wang, Ziang Zhang, Tianyu Pang, Chao Du, Hengshuang Zhao, Zhou Zhao• 2024

Related benchmarks

TaskDatasetResultRank
3D Pose EstimationPASCAL3D+
Mean 30° Acc78.4
9
Category-level Pose EstimationObjectron
Median Error15.1
5
Category-level Pose EstimationImageNet 3D
Median Error25.7
5
3D Orientation EstimationOmniObject3D (out-of-distribution)
Acc @ 10°4.77
5
Category-level Pose EstimationREAL275
Median Error26.7
5
Category-level Pose EstimationOmni6DPose
Median Error (Med)54.5
5
Category-level Pose EstimationAvg. All five datasets
Median Error (Med)27.5
5
Absolute Orientation EstimationImageNet3D (test)
Median Error (Deg)13.34
2
Absolute Orientation EstimationSUN-RGBD (test)
Median Angular Error33.94
2
Absolute Orientation EstimationARKitScenes (test)
Median Error77.58
2
Showing 10 of 13 rows

Other info

Follow for update