princeton-vl / infinigen

Infinite Photorealistic Worlds using Procedural Generation
https://infinigen.org
BSD 3-Clause "New" or "Revised" License
5.21k stars 433 forks source link

Customize Target Output #272

Open ykwongaq opened 2 weeks ago

ykwongaq commented 2 weeks ago

First of all, thank you so much for creating such a great work. It helps me a lot!

When running the code, it will generate a lot of output, such as segmentation mask, depth, etc. For example, is there any way to tell Infinigen that it only need to generate images, depths, and camera intrinsic and extrinsic?

Thank you.

araistrick commented 1 week ago

You can achieve this somewhat by editing passes_to_save to customize blender's output here https://github.com/princeton-vl/infinigen/blob/967970e6cbb3239af6ce87f2e1eb289c2616fb88/infinigen_examples/configs_nature/base.gin#L135.

If you are using opengl_gt I dont believe there is a good mechanism to prevent extra passes from being generated (@lahavlipson (?)), but they do not add significant runtime so you can just delete them once finished for the timebeing.