nv-tlabs / ATISS

Code for "ATISS: Autoregressive Transformers for Indoor Scene Synthesis", NeurIPS 2021
Other
258 stars 55 forks source link

Do a core-ML version #8

Closed julienbelangerunity closed 2 years ago

julienbelangerunity commented 2 years ago

I'd love to be able to try the algorithm with no-rendering and all the madness, I'd love to see a minimal version with only the ML core like Dataset in and torch tensor out.

paschalidoud commented 2 years ago

Hi @julienbelangerunity,

Thanks for your feedback. The "ML core" of our code can be found within the networks folder of our repository. This part of the code is completely independent of rendering etc.

Best, Despoina