search

chenguolin / InstructScene

[ICLR 2024 spotlight] Official implementation of "InstructScene: Instruction-Driven 3D Indoor Scene Synthesis with Semantic Graph Prior".
https://chenguolin.github.io/projects/InstructScene
MIT License
55 stars 7 forks source link

release the dataset preprocessing scripts #6

Closed franklhj closed 3 months ago

franklhj commented 3 months ago

Thank you for your great work! When do you plan to release the dataset preprocessing scripts. I noticed that you have a plan to release.

chenguolin commented 3 months ago

Hi @franklhj,

Sorry for the late reply.

It's kind of tricky to clean the codes for dataset preprocessing, as I wrote them roughly for quick implementation. If you would like to use the preprocessed dataset, you can just download our dataset from Huggingface directly, or you can check the 3D-FRONT preprocessing scripts in ATISS, which are almost the same as what we did.

Additionally, I have uploaded the data preprocessing script that applies OpenAI ChatGPT API for caption refinement here. Hope it could be helpful.