search

sherwinbahmani / cc3d

CC3D: Layout-Conditioned Generation of Compositional 3D Scenes
https://sherwinbahmani.github.io/cc3d
92 stars 3 forks source link

How to get data file .zip and .npz in generate.sh ? #5

Closed dengxuan08 closed 1 year ago

dengxuan08 commented 1 year ago

Thanks for the impressive work.
I want to test the generation with my own data, what data do I need to prepare and how do I preprocess it to get the .zip and .npz files ?

image
sherwinbahmani commented 1 year ago

Hi,

We used this script to render the images: https://github.com/DLR-RM/BlenderProc/tree/main/examples/datasets/front_3d_with_improved_mat To create the boxes.npz in the required format we used the ATISS pre-processing code: https://github.com/nv-tlabs/atiss

You then just zip it with two subfolder for images and labels, having the same subsequence names as subsubfolders

dengxuan08 commented 1 year ago

Thanks for the reply, I use https://github.com/sherwinbahmani/cc3d/issues/6#issuecomment-1703950163 for data preprocessing.