nv-tlabs / ATISS

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

preprocess_data.py ends without generating folders #39

Open Hayleymol opened 1 year ago

Hayleymol commented 1 year ago

Hi! I am trying to run the code on my window machine. The preprocess_data.py code does run and generates the following prints.

Applying threed_front_livingroom filtering Loading dataset 6812 / 6813 Loading dataset with 621 rooms Saving training statistics for dataset with bounds: {'translations': (array([-5.67291869, 0.0375 , -5.71640158]), array([5.09667922, 3.35774051, 5.40485 ])), 'sizes': (array([0.03999 , 0.02000002, 0.0328435 ]), array([2.38027 , 1.770065, 1.322429])), 'angles': (array([-3.14159265]), array([3.14159265]))} to \processed\livingrooms\dataset_stats.txt Applying threed_front_livingroom filtering Loading dataset 6812 / 6813 {'translations': (array([-5.67291869, 0.0375 , -5.71640158]), array([5.09667922, 3.35774051, 5.40485 ])), 'sizes': (array([0.03999 , 0.02000002, 0.02799703]), array([2.38027 , 1.770065 , 1.4137885])), 'angles': (array([-3.14159265]), array([3.14159265]))} Loading dataset with 813 rooms 0it [00:00, ?it/s]

13it [00:00, 128.06it/s]

...

813it [00:06, 116.71it/s]

the number of iterations goes to 813 (the number of the living room) but ends without producing any processed files including the folder.

Anyone who faced similar issues? It would be great to get some ideas.