Skip to content

Latest commit

 

History

History
128 lines (111 loc) · 3.77 KB

DATASET.md

File metadata and controls

128 lines (111 loc) · 3.77 KB

Dataset

The overall directory structure should be:

│Point-MAE/
├──cfgs/
├──data/
│   ├──ModelNet/
│   ├──ModelNetFewshot/
│   ├──ScanObjectNN/
│   ├──ScanObjectNN_hardest_fewshot/
│   ├──ShapeNet55-34/
│   ├──shapenetcore_partanno_segmentation_benchmark_v0_normal/
│   ├──Stanford3dDataset_v1.2_Aligned_Version/
│   ├──stanford_indoor3d/
├──datasets/
├──.......

ModelNet40 Dataset:

│ModelNet/
├──modelnet40_normal_resampled/
│  ├── modelnet40_shape_names.txt
│  ├── modelnet40_train.txt
│  ├── modelnet40_test.txt
│  ├── modelnet40_train_8192pts_fps.dat
│  ├── modelnet40_test_8192pts_fps.dat

Download: You can download the processed data from Point-BERT repo, or download from the official website and process it by yourself.

ModelNet Few-shot Dataset:

│ModelNetFewshot/
├──5way10shot/
│  ├── 0.pkl
│  ├── ...
│  ├── 9.pkl
├──5way20shot/
│  ├── ...
├──10way10shot/
│  ├── ...
├──10way20shot/
│  ├── ...

Download: Please download the data from Point-BERT repo. We use the same data split as theirs.

ScanObjectNN Few-shot Dataset:

│ScanObjectNN_hardest_fewshot/
├──5way10shot/
│  ├── 0.pkl
│  ├── ...
│  ├── 9.pkl
├──5way20shot/
│  ├── ...
├──10way10shot/
│  ├── ...
├──10way20shot/
│  ├── ...

Download: Please download the data from MaskSurf_GoogleDrive.

ScanObjectNN Dataset:

│ScanObjectNN/
├──main_split/
│  ├── training_objectdataset_augmentedrot_scale75.h5
│  ├── test_objectdataset_augmentedrot_scale75.h5
│  ├── training_objectdataset.h5
│  ├── test_objectdataset.h5
├──main_split_nobg/
│  ├── training_objectdataset.h5
│  ├── test_objectdataset.h5

Download: Please download the data from the official website.

ShapeNet55/34 Dataset:

│ShapeNet55-34/
├──shapenet_pc_masksurf_with_normal/
│  ├── 02691156-1a04e3eab45ca15dd86060f189eb133.npy
│  ├── 02691156-1a6ad7a24bb89733f412783097373bdc.npy
│  ├── .......
├──ShapeNet-55/
│  ├── train.txt
│  └── test.txt

Download: Please download the data from MaskSurf_GoogleDrive.

ShapeNetPart Dataset:

|shapenetcore_partanno_segmentation_benchmark_v0_normal/
├──02691156/
│  ├── 1a04e3eab45ca15dd86060f189eb133.txt
│  ├── .......
│── .......
│──train_test_split/
│──synsetoffset2category.txt

Download: Please download the data from here.

S3DIS Dataset:

|Stanford3dDataset_v1.2_Aligned_Version/
├──Area_1/
│  ├── conferenceRoom_1
│  ├── .......
│── .......
│stanford_indoor3d
│──Area_1_conferenceRoom_1.npy
│──Area_1_office_19.npy

Please prepare the dataset following PointNet: download the Stanford3dDataset_v1.2_Aligned_Version from here, and get the processed stanford_indoor3d with:

cd data_utils
python collect_indoor3d_data.py