CVMI-Lab / PLA

(CVPR 2023) PLA: Language-Driven Open-Vocabulary 3D Scene Understanding & (CVPR2024) RegionPLC: Regional Point-Language Contrastive Learning for Open-World 3D Scene Understanding
Apache License 2.0
256 stars 11 forks source link

scripts to generate the captions for S3DIS dataset #29

Closed Areeba-Raza closed 5 months ago

Areeba-Raza commented 11 months ago

Hello ! Can you please provide the link for the scripts for the generation of the captions for S3DIS dataset. because I beileve they are not implemented in the scripts that you shared in Dataset.md file.

I look forward to your response

Thanks

Dingry commented 11 months ago

This is the sample code for creating captions and caption indices for ScanNet. It can be easily modified to adapt to S3DIS.

https://github.com/CVMI-Lab/PLA/blob/dev/tools/process_tools/generate_caption.py#L421-L443 https://github.com/CVMI-Lab/PLA/blob/dev/tools/process_tools/generate_caption_idx.py#L224-L241