README.md 951 Bytes
Newer Older
liyinhao's avatar
liyinhao committed
1
### Prepare ScanNet Data
2
We follow the procedure in [votenet](https://github.com/facebookresearch/votenet/).
liyinhao's avatar
liyinhao committed
3

liyinhao's avatar
liyinhao committed
4
1. Download ScanNet v2 data [HERE](https://github.com/ScanNet/ScanNet). Link or move the 'scans' folder to this level of directory.
liyinhao's avatar
liyinhao committed
5

6
2. In this directory, extract point clouds and annotations by running `python batch_load_scannet_data.py`.
7

8
9
10
11
3. Enter the project root directory, generate training data by running
```bash
python tools/create_data.py scannet --root-path ./data/scannet --out-dir ./data/scannet --extra-tag scannet
```
liyinhao's avatar
liyinhao committed
12

13
The directory structure after pre-processing should be as below
liyinhao's avatar
liyinhao committed
14
15
16
```
scannet
├── scannet_utils.py
17
├── batch_load_scannet_data.py
liyinhao's avatar
liyinhao committed
18
├── load_scannet_data.py
19
├── scannet_utils.py
liyinhao's avatar
liyinhao committed
20
21
├── README.md
├── scans
22
23
24
25
26
27
├── scannet_train_instance_data
├── points
├── instance_mask
├── semantic_mask
├── scannet_infos_train.pkl
├── scannet_infos_val.pkl
liyinhao's avatar
liyinhao committed
28
29

```