Skip to content

Latest commit

 

History

History
92 lines (79 loc) · 5.41 KB

README.md

File metadata and controls

92 lines (79 loc) · 5.41 KB
Raw

Zeroverse

Official code for NeurIPS 2024 paper LRM-Zero: Training Large Reconstruction Models with Synthesized Data, by Desai Xie, Sai Bi, Zhixin Shu, Kai Zhang, Zexiang Xu, Yi Zhou, Soren Pirk, Arie Kaufman, Xin Sun, Hao Tan, a collaboration between Adobe Research, Stony Brook University, and Kiel University. lrmzero_zeroverse_figure_v2

arxiv project website

This repository only contains the minimal code for procedurally synthesizing a single Zeroverse object. To generate a large-scale Zeroversedataset, you need to parallelly run the code with different seed for each object. create_shapes.py is based on the SIGGRAPH 2018 (TOG) paper Deep Image-Based Relighting from Optimal Sparse Samples (PDF, Project Webpage, GitHub link).

Here we use a publicly available material dataset, MatSynth, which is different from the internal material dataset that we used for all experiment results reported in the paper. This code is not extensively tested after switching to MatSynth, so please let us know if you encounter any issues.

Environment Setup

  • requires python3.10
sudo apt install libxi6 libsm6 libxext6
pip install datasets opencv-python pillow rich bpy==3.6.0
# install blender, then
export PATH=path/to/blender/:$PATH  # needs blender binary in addition to bpy to run zeroverse_rgba.py, augment_shapes.py and 2gltf2.py

Randomly Sampled Parameters for Generated Shapes

  • create_shapes.py
    • uuid: str
    • sub_obj_num: int, number of sub objects of this shape
    • max_dim: float
    • sub_objs: [{}, {}, ...]
      • primitive_id: int, range(3)
      • axis_vals: array (3,)
      • translation: array (3,)
      • translation1: array (3,)
      • rotation: array (3,)
      • rotation1: array (3,)
      • height_fields: array (6, 36, 36)
      • material_id: int, rgb2x material id
  • augment_shapes.py
    • boolean augmentation
      • is_augmented: boolean
      • size: random.uniform(0.3, 0.6), size of cutter
      • pos_0: mathutils.Vector of size 3, coordinate of a random vertex of the existing shape, used as the center of the cutter
      • cut_type: random.choice(['sphere', 'cube', 'cone', 'cylinder', 'torus'])
      • rotation_euler: (random.uniform(0, 360) for _ in range(3))
      • thickness: random.uniform(0.01, 0.02), thickness of solidify after cutting
    • wireframe augmentation
      • is_wireframed: boolean
      • size: random.uniform(0.3, 0.6), size of wireframe primitive
      • thickness: random.uniform(0.01, 0.03), thickness of wireframe
      • wire_type: random.choice(['sphere', 'cube', 'cone', 'cylinder', 'torus'])
      • subdivide_type: random.choice(['SIMPLE', 'CATMULL_CLARK'])
      • subdivide_level: random.choice([0, 1])
      • pos_0: mathutils.Vector of size 3, coordinate of a random vertex of the existing shape, used as the center of the cutter
      • rotation_euler: (random.uniform(0, 360) for _ in range(3))
      • material_name: str, name of a material in the existing shape

Synthesizing and Rendering a Single Object

  1. Synthesize objects in .obj and .mtl and convert to .glb: python create_shapes.py --seed 0 --output_dir outputs/
  2. Optionally add boolean difference or wireframe augmentation and render .glb: python zeroverse_rgba.py --seed 0 --object_path outputs/some_uuid/some_uuid.glb --output_dir outputs/some_uuid/views
    • add --boolean_probability 1.0 and/or --wireframe_probability 1.0 to add these augmentations, which will be done right before rendering the views.

Example of a boolean-difference augmented object:

Zeroverse Dataset Generation

  • You just need to parallelize the above single-shape synthesizing process and run it for 800K times, each object with a different seed. Then you will get your own Zeroverse dataset with 800K objects, matching the size of Objaverse!

Citation

If you find this code useful, please consider citing:

@misc{xie2024lrmzero,
      title={LRM-Zero: Training Large Reconstruction Models with Synthesized Data}, 
      author={Desai Xie and Sai Bi and Zhixin Shu and Kai Zhang and Zexiang Xu and Yi Zhou and Sören Pirk and Arie Kaufman and Xin Sun and Hao Tan},
      year={2024},
      eprint={2406.09371},
      archivePrefix={arXiv},
      primaryClass={id='cs.CV' full_name='Computer Vision and Pattern Recognition' is_active=True alt_name=None in_archive='cs' is_general=False description='Covers image processing, computer vision, pattern recognition, and scene understanding. Roughly includes material in ACM Subject Classes I.2.10, I.4, and I.5.'}
}

@article{xu2018deep,
  title={Deep image-based relighting from optimal sparse samples},
  author={Xu, Zexiang and Sunkavalli, Kalyan and Hadap, Sunil and Ramamoorthi, Ravi},
  journal={ACM Transactions on Graphics (TOG)},
  volume={37},
  number={4},
  pages={126},
  year={2018},
  publisher={ACM}
}