Skip to content

Latest commit

 

History

History
48 lines (41 loc) · 3.04 KB

DATASET.md

File metadata and controls

48 lines (41 loc) · 3.04 KB
Raw

The dataset configs are located within tools/cfgs/dataset_configs, and the model configs are located within tools/cfgs for different settings.

ScanNet Dataset

  • Please download the ScanNet Dataset and follow PointGroup to pre-process the dataset as follows or directly download the pre-processed data here.

  • Additionally, please download the caption data here. If you want to generate captions on your own, please download image data (scannet_frames_25k) from ScanNet and follow scripts generate_caption.py and generate_caption_idx.py.

  • The directory organization should be as follows:

    PLA
├── data
│   ├── scannetv2
│   │   │── train
│   │   │   │── scene0000_00.pth
│   │   │   │── ...
│   │   │── val
│   │   │── text_embed
│   │   │── caption_idx
│   │   │── scannetv2_train.txt
│   │   │── scannetv2_val.txt
│   │   │—— scannet_frames_25k (optional, only for caption generation)
├── pcseg
├── tools

S3DIS Dataset

  • Please download the S3DIS Dataset and follow dataset/s3dis/preprocess.py to pre-process the dataset as follows or directly download the pre-processed data here.

    python3 pcseg/datasets/s3dis/preprocess.py

  • Additionally, please download the caption data here. If you want to generate captions on your own, please download image data here and follows scripts here: generate_caption.py and generate_caption_idx.py.

  • The directory organization should be as follows:

    PLA
├── data
│   ├── s3dis
│   │   │── stanford_indoor3d_inst
│   │   │   │── Area_1_Conference_1.npy
│   │   │   │── ...
│   │   │── text_embed
│   │   │── caption_idx
│   │   │—— s3dis_2d (optional, only for caption generation)
├── pcseg
├── tools