README.md 1.15 KB
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

liyinhao's avatar
liyinhao committed
13
14
15
16
17
18
19
The overall process could be achieved through the following script
```bash
python batch_load_scannet_data.py
cd ../..
python tools/create_data.py scannet --root-path ./data/scannet --out-dir ./data/scannet --extra-tag scannet
```

20
The directory structure after pre-processing should be as below
liyinhao's avatar
liyinhao committed
21
22
23
```
scannet
├── scannet_utils.py
24
├── batch_load_scannet_data.py
liyinhao's avatar
liyinhao committed
25
├── load_scannet_data.py
26
├── scannet_utils.py
liyinhao's avatar
liyinhao committed
27
28
├── README.md
├── scans
29
30
31
32
33
34
├── scannet_train_instance_data
├── points
├── instance_mask
├── semantic_mask
├── scannet_infos_train.pkl
├── scannet_infos_val.pkl
liyinhao's avatar
liyinhao committed
35
36

```