This project offers the Canonical Objaverse Dataset, created using the methods outlined in the paper "One-shot 3D Object Canonicalization based on Geometric and Semantic Consistency." Additionally, the project provides an Objaverse toolkit that includes functionalities for downloading Objaverse data, converting GLB files to OBJ format, loading canonical poses, and more.
- Release COD dataset and toolkit.
- Release paper on arxiv.
The code has been tested with
- python 3.10
- pytorch 1.9.0
- pytorch3d 0.7.5
- open3d 0.14.1
├── Example
│ ├── load_canonicalData.py # A pipeline that includes downloads, format conversion, and canonicalization
├── Src
│ ├── DataDownload
│ │ ├── Download.py # Download Objaverse Data from internet
│ │ └── ..
│ ├── Glb2Obj
│ │ ├── glb2Obj.py # Convert GLB files to OBJ format
│ │ └── ..
│ └── Obj2Canonicalization
│ │ ├── canon_obj.py # Load canonical labels on object canonicalization
│ │ └── ..
│ └── structure
│ │ ├── ..
│ └── utils
│ │ ├── ..
├── data
├── CanonicalObjaverseDataset.json # Canonicalization labels
└── canon-annotations.json # Category and object's UID
To test the model, please run:
python -m Examples.load_canonicalData --data_name 'test'
To get all the COD datasets, please run:
python -m Examples.load_canonicalData --data_name 'canon'
Coming soon.
This work was completed during a visit to the Visual Computing and Learning Laboratory at Peking University. Special thanks to Prof. Baoquan Chen and the co-authors for their support. Grateful acknowledgment is extended to Prof. Pengshuai Wang, Prof. Xifeng Gao, and Dr. Siyan Dong for their guidance and discussions. Thanks also to Kai Ye, Jia Li, and Yuhang He for their assistance.